2017年多校赛第九场 1005 FFF at Valentine（缩点+拓扑排序）

其实缩点很容易想到，就是不怎么会用拓扑排序所以一直卡着判断这个地方。

代码如下：

#include <stdio.h>
#include <string.h>
#include <vector>
#include <stack>
#include <algorithm>
#include <iostream>
#include<queue>
using namespace std;
const int N = 1005; // 点的最大个数
stack <int> sta; // 存放遍历了的点
vector <int> edge
; // 记录每个点能到达哪些点
int dfn
; // 第几个遍历
int low
; // 根是多少
int key
; // 三种状态，在队列为2，不在队列为0，已经访问但不在栈中为1
vector <int> com
; // 每个com里面都是一个强联通分量，最多有N个
int incom
; // 记录每个点在第几个强联通分量里面
int ind, com_num; // 下标， 强联通个数
int n, m; // 点数和边数
int in
;
vector <int> G
;

void init()
{
memset(dfn, -1, sizeof(dfn));
memset(low, -1, sizeof(low));
memset(key, 0, sizeof(key));
memset(in, 0, sizeof(in));
ind = com_num = 0;
for(int i = 0; i <= n; i++)
{
edge[i].clear();
G[i].clear();
}

while(!sta.empty())
sta.pop();
}

void tarjan(int u)
{
int v;
key[u] = 2; //放入栈的标记
low[u] = dfn[u] = ind++; //u是第ind个被遍历到
sta.push(u); //遍历到的加入栈中

for(int i = 0; i < edge[u].size(); i++)
{
v = edge[u][i];
if(dfn[v] == -1) //v这个点还未遍历到，则遍历
{
tarjan(v);
low[u] = min(low[u], low[v]); //取能遍历到的最小的根序号
}
else if(key[v] == 2) //v这个点已经遍历到且仍存于栈中，则取他们遍历顺序的较小值
low[u] = min(low[u], dfn[v]);
}

if(low[u] == dfn[u]) //如果某个点遍历下去又回到了这个u，即为强联通，会使得该情况成立，则有
{
while(!sta.empty()) //把栈中的点取出，直至取到u
{
int j = sta.top();
sta.pop();
key[j] = 1; //这个点已经遍历过且不需再遍历
incom[j] = com_num; //记录下j在第com个强连通分量集合里面
if (j == u)
break;
}
com_num++; //一个强联通分量产生
}
}

queue <int> q;

int main()
{
int T;
scanf("%d", &T);
int a, b, m;
while(T--)
{
scanf("%d%d", &n, &m);
init();
for(int i = 1; i <= m; i++)
{
scanf("%d%d", &a, &b);
edge[a].push_back(b);
}
for(int i = 1; i <= n; i++)
if(dfn[i] == -1)
tarjan(i);

for(int i = 1; i <= n; i++) {
for(int j = 0; j < edge[i].size(); j++) {
int v = edge[i][j];
if(incom[i] == incom[v])
continue;
G[incom[i]].push_back(incom[v]);
in[incom[v]]++;
}
}

for(int i = 0; i < com_num; i++)
if(in[i] == 0)
q.push(i);
bool flag = 0;
while(!q.empty()) {
if(q.size() > 1)
flag = 1;
int u = q.front();
q.pop();
for(int i = 0; i < G[u].size(); i++) {
int v = G[u][i];
in[v]--;
if(in[v] == 0)
q.push(v);
}
}
if(!flag) {
printf("I love you my love and our love save us!\n");
} else {
printf("Light my fire!\n");
}
}
return 0;
}