bzoj 5091: 摘苹果 期望+乱搞

题意

小Q的工作是采摘花园里的苹果。在花园中有n棵苹果树以及m条双向道路，苹果树编号依次为1到n，每条道路的两
端连接着两棵不同的苹果树。假设第i棵苹果树连接着d_i条道路。小Q将会按照以下方式去采摘苹果：
1.小Q随机移动到一棵苹果树下，移动到第i棵苹果树下的概率为d_i/(2m)，但不在此采摘。
2.等概率随机选择一条与当前苹果树相连的一条道路，移动到另一棵苹果树下。
3.假设当前位于第i棵苹果树下，则他会采摘a_i个苹果，多次经过同一棵苹果树下会重复采摘。
4.重复第2和3步k次。
请写一个程序帮助计算小Q期望摘到多少苹果。
n,k<=100000,m<=200000

分析

比赛的时候只想着如何优化暴力递推，就没有搞出来。
这题要注意到题目给出的特殊条件：移动到第i棵苹果树的概率是$\frac{d_i}{2m}$，假设有一条i到j的边，那么在第一步到达j的概率就要加上$\frac{1}{2m}$。因为要加$d_j$次，所以第1步到达j的概率就是$\frac{d_j}{2m}$。同理可以得到每一步到达点j的概率其实就是$\frac{d_j}{2m}$。
所以答案就是$k*\sum_{i=1}^n\frac{d_i*a_i}{2m}$

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=100005;
const int MOD=1000000007;

int n,m,k,a[N],d[N];

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 ksm(int x,int y)
{
    int ans=1;
    while (y)
    {
        if (y&1) ans=(LL)ans*x%MOD;
        x=(LL)x*x%MOD;y>>=1;
    }
    return ans;
}

int main()
{
    n=read();m=read();k=read();
    for (int i=1;i<=n;i++) a[i]=read();
    for (int i=1;i<=m;i++)
    {
        int x=read(),y=read();
        d[x]++;d[y]++;
    }
    int ans=0,ny=ksm(m*2,MOD-2);
    for (int i=1;i<=n;i++) ans+=(LL)d[i]*ny%MOD*a[i]%MOD,ans-=ans>=MOD?MOD:0;
    ans=(LL)ans*k%MOD;
    printf("%d",ans);
    return 0;
}