1113 - 这一定不是C

DESCRIPTION

有一棵$N$个点的树，显然这棵树上共有$\frac{n\times (n+1)}{2}$ 条不同的简单路径

现在求有多少个不同的简单路径$K$元组$(P_1,P_2,....,P_K)$， 使得任意两条$K$元组中的路径有至少一个公共点

注意， 一条简单路径可以出现在$K$元组中多次

INPUT

第一行两个整数 $N,K$接下来 $N-1$个数， 第 $i$个数表示以 $1$号点为根的情况下， 第 $i+1$个节点的父节点编号 $F_{i+1}$， 保证 $1\le F_{i+1}\le i$

OUTPUT

一行一个整数表示答案对 ${10}^9+7$取模的结果

SAMPLE INPUT

2 21

SAMPLE OUTPUT

7

HINT

$1\le N\le 5\times {10}^5$, $1\le K\le {10}^9$对于样例， 只有{(1,1),(2,2)}， {(2,2),(1,1)}不合法

SOLUTION

“玲珑杯”ACM比赛 Round #13

C

注意到， $K$条路径两两相交的充要条件是这$K$条路径有一段公共路径。

对于一条路径， 点数-边数恒为1

所以我们可以算出所有可能的情况中， $K$条路径的公共点的个数 减去$K$条路径的公共边的条数， 即为答案

我们可以枚举每一个点以及每一条边的贡献， 对于一个点来说， 它的贡献就是所有经过它的路径条数的$K$次方，边同理

复杂度$O\left(NlogK\right)$， 瓶颈在于快速幂

这个题根本坐不上，看了题解还理解了半天，确实是思想不到位，最开始搜索解决。爆栈了，可能就是有一条链的情况。

最后还是改成了dp来做。就可以了。主要说找点经过的路径，边好找一些。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const int maxn=1e6;
ll n;
ll dp[maxn+10];
ll k;
vector<int>ve[maxn+10];
ll mul(ll a,ll b)
{
    ll ret=1;
    while(b)
    {
        if(b&1)
            ret=ret*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ret;
}
int fa[maxn+10];
int main()
{
    while(~scanf("%lld%lld",&n,&k))
    {
        for(int i=0;i<=maxn;i++)
        {
            ve[i].clear();
            dp[i]=0;
        }
        for(int i=2;i<=n;i++)
        {
            scanf("%d",&fa[i]);
            ve[fa[i]].push_back(i);
        }
        for(int i=n;i>=1;i--)
        {
            dp[fa[i]]+=dp[i]+1;
        }
//        for(int i=1;i<=n;i++)
//            printf("%d\n",dp[i]);
        ll ans=0;
        ll ans1=0;
        for(int i=1;i<=n;i++)
        {
//            ans=(ans+mul(dp[i]*(n-1-dp[i]),k))%mod;
//            ans1=(ans1+(dp[i]+1)*(n-dp[i]-1))%mod;

                ll num=0;
                ll sum=0;
                for(int j=0;j<ve[i].size();j++)
                {
                    int v=ve[i][j];
                    num=(num+sum*(dp[v]+1)%mod)%mod;
                    sum=(sum+(dp[v]+1))%mod;
                }
//
                ll tmp=dp[i]*(n-1-dp[i])%mod;
                ans=(ans+mul((tmp+num+n)%mod,k))%mod;

                //ll tmp=(num+dp[i]+1)%mod;
                ll t=(n-dp[i]-1)*(dp[i]+1)%mod;
                //ans=(ans+mul((tmp+t)%mod,k))%mod;
                ans1=(ans1+mul(t,k))%mod;
        }
        printf("%lld\n",((ans-ans1)%mod+mod)%mod);
    }
    return 0;
}