BZOJ 1589 [Usaco2008 Dec] Tarjan缩点+记忆化搜索 解题报告

1589: [Usaco2008 Dec]Trick or Treat on the Farm 采集糖果

Description

每年万圣节，威斯康星的奶牛们都要打扮一番，出门在农场的N(1≤N≤100000)个牛棚里转悠，来采集糖果．她们每走到一个未曾经过的牛棚，就会采集这个棚里的1颗糖果． 农场不大，所以约翰要想尽法子让奶牛们得到快乐．他给每一个牛棚设置了一个“后继牛棚”．牛棚i的后继牛棚是Xi．他告诉奶牛们，她们到了一个牛棚之后，只要再往后继牛棚走去，就可以搜集到很多糖果．事实上这是一种有点欺骗意味的手段，来节约他的糖果． 第i只奶牛从牛棚i开始她的旅程．请你计算，每一只奶牛可以采集到多少糖果．

Input

第1行输入N，之后一行一个整数表示牛棚i的后继牛棚Xi，共N行．

Output

共N行，一行一个整数表示一只奶牛可以采集的糖果数量．

Sample Input

4 //有四个点
1 //1有一条边指向1
3 //2有一条边指向3
2 //3有一条边指向2
3

Sample Output

1
2
2
3

【解题报告】

为什么我一开始要去写边表啊。。。
又臭又长。。。
记录一个后继就可以了

代码如下：

/**************************************************************
    Problem: 1589
    User: onepointo
    Language: C++
    Result: Accepted
    Time:728 ms
    Memory:46972 kb
****************************************************************/

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 1000010

int n,cnt,num,indexx,head[N],head_[N];
int to[N],nxt[N];
int dfn[N],low[N];
int st[N],top,ins[N];
int belong[N],size[N];
int dp[N];

void init()
{
    cnt=0;num=indexx=top=0;
    memset(to,0,sizeof(to));
    memset(nxt,0,sizeof(nxt));
    memset(dfn,-1,sizeof(dfn));  
    memset(low,-1,sizeof(low));    
    memset(ins,0,sizeof(ins));  
    memset(belong,0,sizeof(belong));
    memset(size,0,sizeof(size));
}
void tarjan(int u)
{
    dfn[u]=low[u]=++num;
    ins[u]=1;st[++top]=u;
    int v=nxt[u];
    if(dfn[v]==-1){tarjan(v);low[u]=min(low[u],low[v]);}
    else if(ins[v]){low[u]=min(low[u],dfn[v]);}
    if(low[u]==dfn[u])
    {
        indexx++;int v=0;
        while(v!=u)
        {
            v=st[top--];ins[v]=0;
            belong[v]=indexx;
            size[indexx]++;
        }
    }
}
int dfs(int u)
{
    if(dp[u]) return dp[u];
    dp[u]=size[u];
    if(to[u]) dp[u]+=dfs(to[u]);
    return dp[u];
}
int main()
{
    init();
    scanf("%d",&n);
    for(int i=1;i<=n;++i) scanf("%d",&nxt[i]);
    for(int i=1;i<=n;++i) if(dfn[i]==-1) tarjan(i);
    for(int i=1;i<=n;++i) if(belong[i]!=belong[nxt[i]]) to[belong[i]]=belong[nxt[i]];
    for(int i=1;i<=n;++i) printf("%d\n",dfs(belong[i]));
    return 0;
}