题目描述
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);
}