人生的意义

题目描述

NiroBC在温暖的YYHS衣来伸手饭来张口，过着颓废的生活。
终于有一天，她意识到，真正有意义的人生不应该在如此空虚的时光中度过，于是她下定决心，来到了百废待兴的新世界，踏上了寻找人生的意义的旅途。
这个百废待兴的新世界啊真是美妙，绽放着令人惊喜的异彩，NiroBC欢笑着投入了新世界的怀抱，随机的降落在这个新世界中。
这个新世界由N个城市组成(编号为1..N)，每个城市中都能找到特定价值的人生意义，其中第i个城市的意义大小为W[i]。
NiroBC渐渐熟悉了这个世界的规律，她发现人世间发生的一切都是注定的，不可能因为世界的一个小部分————也就是人的个体————的意志而改变。既然一切都是注定的，那么每个城市都有一个唯一的后继城市a[i]。也就是说，当NiroBC当前在城市i时，下一时刻她将不可选择地来到a[i]。
设NiroBC最初降落在城市x，那么她将按照x -> a[x] -> a[a[x]] -> a[a[a[x]]] -> …的路径行走，永远不会停止，并且得到沿途的人生意义。然而人生不能单调，同一个城市的人生意义只能得到一次，NiroBC最终得到的人生意义是所有经过的城市(不重复)的人生意义的和。
NiroBC是等概率地随机地降落在N个城市中的，她想知道她最终得到的人生的意义的期望值。

水题

搜一下呗。
用了栈来弄环。
不要递归。

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
typedef long long ll;
typedef double db;
const int maxn=300000+10;
int a[maxn],co[maxn],f[maxn],w[maxn],sta[maxn];
int bz[maxn];
int i,j,k,l,r,t,n,m,tot,top;
ll ans;
int read(){
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9'){
        if (ch=='-') f=-1;
        ch=getchar();
    }
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
int main(){
    freopen("meaning.in","r",stdin);freopen("meaning.out","w",stdout);
    n=read();
    fo(i,1,n) a[i]=read();
    fo(i,1,n) w[i]=read();
    t=0;
    fo(i,1,n)
        if (!bz[i]){
            ++t;
            top=0;
            k=i;
            while (1){
                bz[k]=t;
                sta[++top]=k;
                if (bz[a[k]]){
                    if (bz[a[k]]==t){
                        l=0;r=top;
                        while (1){
                            l+=w[sta[top]];
                            top--;
                            if (sta[top+1]==a[k]) break;
                        }
                        fo(j,top+1,r) f[sta[j]]=l;
                    }
                    else{
                        f[k]=f[a[k]]+w[k];
                        top--;
                    }
                    break;
                }
                k=a[k];
            }
            while (top){
                f[sta[top]]=f[sta[top+1]]+w[sta[top]];
                top--;
            }
        }
    fo(i,1,n) ans+=(ll)f[i];
    printf("%.6lf\n",(db)ans/n);
}