對于一個牛，它存在兩種狀態(tài)：1.處于聯(lián)通分量 2.不處于聯(lián)通分量。對于處于聯(lián)通分量的牛，求出聯(lián)通分量的大小；對于不處于聯(lián)通分量的牛，求出其距離聯(lián)通分量的路程+聯(lián)通分量大小。

不同的聯(lián)通分量，染上不同的顏色，可以計算各個聯(lián)通分量的大小。

#include<bits/stdc++.h> using namespace std;int vis[100005]; int low[100005]; int Loop[100005],color[100005],dfn[100005],head[100005],ans[100005],st[100005], Next[100005]; int pos; int k=1; int col; int top;struct node {int v,next; }edge[100005];void Add(int u,int v) {edge[k].v = v;edge[k].next = head[u];head[u] = k++; }void Find(int root,int x,int step) {if(ans[x]!=0){ans[root] = ans[x]+step;return;}elseFind(root,Next[x],step+1); }void Tarjan(int u) {dfn[u] = low[u] = ++pos;st[++top] = u;vis[u] = 1;for(int i=head[u];i!=-1;i=edge[i].next){int v = edge[i].v;if(!dfn[v]){Tarjan(v);low[u] = min(low[u],low[v]);}else if(vis[v]){low[u] = min(low[u],dfn[v]);}}if(dfn[u] == low[u]){col++;int x;do{x = st[top--];vis[x] = 0;color[x] = col;}while(x!=u);} }int main() {memset(head,-1,sizeof(head));int n,v;scanf("%d",&n);for(int i=1;i<=n;i++){scanf("%d",&v);Next[i] = v;Add(i,v);if(v==i)ans[i] = 1;}for(int i=1;i<=n;i++){if(!dfn[i]){Tarjan(i);}}for(int i=1;i<=n;i++){Loop[color[i]]++;}for(int i=1;i<=n;i++){if(Loop[color[i]]!=1)ans[i] = Loop[color[i]];}for(int i=1;i<=n;i++){if(ans[i]==0)Find(i,Next[i],1);}for(int i=1;i<=n;i++){printf("%d\n",ans[i]);}return 0; } View Code

?

轉(zhuǎn)載于:https://www.cnblogs.com/alan-W/p/10756744.html

總結(jié)

以上是生活随笔為你收集整理的P2921 [USACO08DEC]在农场万圣节Trick or Treat on the Farm的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。