HDU 3639 Hawk-and-Chicken 強連通分量分解 + dfs

題目：http://acm.hdu.edu.cn/showproblem.php?pid=3639

題意：大概意思是n個人投票，求獲勝者得到多少票，并輸出所有獲勝者，其中支援關系是可以傳遞的，即a -> b, b -> c, c -> d,那麼有a->d

思路：求強連通分量縮點，代表最終獲勝的人所在強連通分量代的點出度必定為0，剩下的就是統計了，然後就比較坑了，剛開始我用的拓撲排序去統計，結果GG了，于是去找資料，結果發現在這裡用拓撲排序搞不定，最後百度了一下：縮點後反向建圖，那麼最終獲勝的人所在強連通分量代的點入度為0，于是以入度為0的點為起點dfs去統計支援他的人數。至于為什麼反向建圖，因為正向建圖統計支援者太坑，反向隻需dfs一遍統計即可。

總結：剛開始看着好簡單。。。結果是個坑。。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
typedef long long ll;
const int N = 5010;
const int INF = 0x3f3f3f3f;
struct edge
{
    int to, next;
} G[N*10], e;
int dfn[N], low[N], scc[N], st[N], ans[N];
int head[N], cost[N];
int index, cnt, num, top;
bool vis[N], used[N];
vector<int> gra[N];
int n, m, sum = 0, x = 0;
void init()
{
    memset(head, -1, sizeof head);
    memset(dfn, -1, sizeof dfn);
    memset(vis, 0, sizeof vis);
    memset(ans, 0, sizeof ans);
    memset(cost, 0, sizeof cost);
    index = cnt = num = top = 0;
}
void add_edge(int v, int u)
{
    G[cnt].to = u;
    G[cnt].next = head[v];
    head[v] = cnt++;
}
void tarjan(int v)
{
    dfn[v] = low[v] = index++;
    vis[v] = true;
    st[top++] = v;
    int u;
    for(int i = head[v]; i != -1; i = G[i].next)
    {
        u = G[i].to;
        if(dfn[u] == -1)
        {
            tarjan(u);
            low[v] = min(low[v], low[u]);
        }
        else if(vis[u])
            low[v] = min(low[v], dfn[u]);
    }
    if(dfn[v] == low[v])
    {
        num++;
        do
        {
            u = st[--top];
            vis[u] = false;
            scc[u] = num;
            ans[num]++;
        }
        while(u != v);
    }
}
void dfs(int v)
{
    used[v] = true;
    sum += ans[v];
    for(int i = 0; i < gra[v].size(); i++)
        if(! used[gra[v][i]])
            dfs(gra[v][i]);
}
void slove()
{
    for(int i = 0; i < n; i++)
        if(dfn[i] == -1)
            tarjan(i);

    int indeg[N];
    memset(indeg, 0, sizeof indeg);
    for(int i = 0; i <= num; i++)
        gra[i].clear();
    for(int i = 0; i < n; i++)
        for(int j = head[i]; j != -1; j = G[j].next)
            if(scc[i] != scc[G[j].to])
            { /*反向建圖*/
                indeg[scc[i]]++;
                gra[scc[G[j].to]].push_back(scc[i]);
            }
    int Max = -1;
    for(int i = 1; i <= num; i++)
        if(indeg[i] == 0) /*統計入度為0的點的支援者*/
        {
            memset(used, 0, sizeof used);
            sum = 0;
            dfs(i);
            cost[i] = sum;
            Max = max(Max, sum);
        }

    int k = 0, res[N];
    for(int i = 1; i <= num; i++)
        if(cost[i] == Max) /*統計獲勝者*/
            for(int j = 0; j < n; j++)
                if(scc[j] == i)
                    res[k++] = j;

    sort(res, res + k);
    printf("Case %d: %d\n", ++x, Max - 1);
    for(int i = 0; i < k; i++)
        if(i == 0) printf("%d", res[i]);
        else printf(" %d", res[i]);
    printf("\n");
}
int main()
{
    int t, a, b;
    scanf("%d", &t);
    while(t--)
    {
        init();
        scanf("%d%d", &n, &m);
        for(int i = 0; i < m; i++)
        {
            scanf("%d%d", &a, &b);
            add_edge(a, b);
        }
        slove();
    }

    return 0;
}