P2921 [USACO08DEC]在农场万圣节Trick or Treat on the Farm

题意翻译

题目描述

每年，在威斯康星州，奶牛们都会穿上衣服，收集农夫约翰在N(1<=N<=100，000)个牛棚隔间中留下的糖果，以此来庆祝美国秋天的万圣节。

由于牛棚不太大，FJ通过指定奶牛必须遵循的穿越路线来确保奶牛的乐趣。为了实现这个让奶牛在牛棚里来回穿梭的方案，FJ在第i号隔间上张贴了一个“下一个隔间”Next_i(1<=Next_i<=N)，告诉奶牛要去的下一个隔间；这样，为了收集它们的糖果，奶牛就会在牛棚里来回穿梭了。

FJ命令奶牛i应该从i号隔间开始收集糖果。如果一只奶牛回到某一个她已经去过的隔间，她就会停止收集糖果。

在被迫停止收集糖果之前，计算一下每头奶牛要前往的隔间数（包含起点）。

输入格式

第1行 整数n。

第2行到n+1行 每行包含一个整数 next_i 。

输出格式

n行，第i行包含一个整数，表示第i只奶牛要前往的隔间数。

样例解释

有4个隔间

隔间1要求牛到隔间1

隔间2要求牛到隔间3

隔间3要求牛到隔间2

隔间4要求牛到隔间3

牛1，从1号隔间出发，总共访问1个隔间；

牛2，从2号隔间出发，然后到三号隔间，然后到2号隔间，终止，总共访问2个隔间；

牛3，从3号隔间出发，然后到2号隔间，然后到3号隔间，终止，总共访问2个隔间；

牛4，从4号隔间出发，然后到3号隔间，然后到2号隔间，然后到3号隔间，终止，总共访问3个隔间。

翻译提供者：吃葡萄吐糖

题目描述

Every year in Wisconsin the cows celebrate the USA autumn holiday of Halloween by dressing up in costumes and collecting candy that Farmer John leaves in the N (1 <= N <= 100,000) stalls conveniently numbered 1..N.

Because the barn is not so large, FJ makes sure the cows extend their fun by specifying a traversal route the cows must follow. To implement this scheme for traveling back and forth through the barn, FJ has posted a 'next stall number' next_i (1 <= next_i <= N) on stall i that tells the cows which stall to visit next; the cows thus might travel the length of the barn many times in order to collect their candy.

FJ mandates that cow i should start collecting candy at stall i. A cow stops her candy collection if she arrives back at any stall she has already visited.

Calculate the number of unique stalls each cow visits before being forced to stop her candy collection.

POINTS: 100

每年万圣节，威斯康星的奶牛们都要打扮一番，出门在农场的N个牛棚里转 悠，来采集糖果.她们每走到一个未曾经过的牛棚，就会采集这个棚里的1颗糖果.

农场不大，所以约翰要想尽法子让奶牛们得到快乐.他给每一个牛棚设置了一个“后继牛 棚”.牛棚i的后继牛棚是next_i 他告诉奶牛们，她们到了一个牛棚之后，只要再往后继牛棚走去， 就可以搜集到很多糖果.事实上这是一种有点欺骗意味的手段，来节约他的糖果.

第i只奶牛从牛棚i开始她的旅程．请你计算，每一只奶牛可以采集到多少糖果．

输入输出格式

输入格式：

 

* Line 1: A single integer: N

* Lines 2..N+1: Line i+1 contains a single integer: next_i

 

输出格式：

 

* Lines 1..N: Line i contains a single integer that is the total number of unique stalls visited by cow i before she returns to a stall she has previously visited.

 

输入输出样例

输入样例#1： 复制

4 
1 
3 
2 
3 

输出样例#1： 复制

1 
2 
2 
3 

说明

Four stalls.

* Stall 1 directs the cow back to stall 1.

* Stall 2 directs the cow to stall 3

* Stall 3 directs the cow to stall 2

* Stall 4 directs the cow to stall 3

Cow 1: Start at 1, next is 1. Total stalls visited: 1.

Cow 2: Start at 2, next is 3, next is 2. Total stalls visited: 2. Cow 3: Start at 3, next is 2, next is 3. Total stalls visited: 2. Cow 4: Start at 4, next is 3, next is 2, next is 3. Total stalls visited: 3.

 

思路：很明显的tarjan算法，不过在这个题目中还要使用缩点，简单解释一下，首先是判断环，然后把环内点的个数记录下来，接下来判断每一头牛，如果这头牛在环内，那么该环的大小就是这头牛最多可以得到糖果的数目，如果该牛在环的外面，那么该牛可以得到的糖果的数目就是离他最近的环的大小加上他到这个环的最近的距离，在这里注意一个条件，就是每一个点的出度都是1，那么我们直接递归往下找环就行了，然后将走过的点与环的大小加起来即可，注意不要忘记特判，下面请看代码

 

#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 100010;
int n, t, cnt = 1, total, index, sum;
int dfn[maxn], Next[maxn], Lemon[maxn], ans[maxn], low[maxn], Stack[maxn], head[maxn], vis[maxn], belong[maxn], result[maxn];
struct node{
    int to;
    int next;
}edge[maxn];

//前向星建图
void add(int u, int v) {
    edge[cnt].to = v;
    edge[cnt].next = head[u];
    head[u] = cnt++;
}

//tarjan算法
void tarjan(int u) {
    dfn[u] = low[u] = ++total;
    vis[u] = 1;
    Stack[++index] = u;
    for (int i = head[u]; i != 0; i = edge[i].next) {
        int v = edge[i].to;
        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]) {
        sum++;
        do {
            belong[Stack[index]] = sum;
            vis[Stack[index--]] = 0;
        } while (Stack[index + 1] != u);
    }
}

void search(int u, int v, int total) {
    if (ans[v]) {
        ans[u] = ans[v] + total;//找到环然后更新该点
    } else {
        search(u, edge[head[v]].to, total + 1);//没找到继续递归查找
    }
}

int main () {
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> t;
        add(i, t);
        if (i == t) ans[i] = 1;//特判
    }
    for (int i = 1; i <= n; i++) {
        if (!dfn[i]) tarjan(i);
    }
    for (int i = 1; i <= n; i++) {
        Lemon[belong[i]]++;//记录环的大小
    }
    for (int i = 1; i <= n; i++) {
        if (Lemon[belong[i]] > 1) ans[i] = Lemon[belong[i]];//如果该点在环内，记录该点所在环的大小
    }
    for (int i = 1; i <= n; i++) {
        if (!ans[i]) {
            search(i, edge[head[i]].to, 1);//递归搜索
        }
        cout << ans[i] << endl;
    }
    return 0;
}