【NOIP2016提高A组五校联考4】label

Description

Input

Output

对于每一组数据，输出答案

Sample Input

3
2 2 0
1 2
3 3 2
1 3
1 2
3 3 1
1 2
2 3

Sample Output

4
2
12

Data Constraint

Solution

对于20%和40% 的数据，可以设DP$f[i,j]$表示节点i选的数是j
那么转移就是$f[i,j]=\Pi_{v\in son[i]} (\sum_{1<=y<=m\bigcap |y-k|>=0} f[v,y])$
y的取值是两段区间，所以可以用前缀和搞定这个sigma
40分到手
接下来就比较…………了
首先把f的值表出来
接着发现对于每个节点，他的f值是对称的，即第一个和最后一个相同，第二个和倒数第二个相同，而中间又有一堆一样的
为什么呢？对于一个节点，它的f值与k有关，k有多大，一开始就会有多少不一样的，而中间的转移式是相同的，sigma以后也是相同的，所以是相同的
那么最大只有(n-1)*k个不同的，中间的存一个，后面的不用存，那存储时就不会爆空间了，只要做前缀和时分类讨论就行了

Code

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define N 110
#define ll long long
#define mo 1000000007ll
#define cl(a) memset(a,0,sizeof(a))
int n,k,next[N*10],last[N*10],to[N*10],tot=0;
ll f[N][30100],s[N][30100],nn,m;
using namespace std;
void putin(int x,int y)
{
    next[++tot]=last[x];last[x]=tot;to[tot]=y;
}
ll sum(int x,ll k)
{
    if(k<=nn) return s[x][k];
    if(k<=m-nn) return (s[x][nn]+f[x][nn+1]*(k-nn))%mo;
    return (s[x][nn]-s[x][m-k]+s[x][nn]+(((m-nn-nn)*f[x][nn+1])%mo)+mo)%mo;
}
void dg(int x,int fa)
{
    for(int l=last[x];l;l=next[l])
    {
        int y=to[l];if(y==fa) continue;
        dg(y,x);
        fo(i,1,nn+1)
        {
            ll jy=0ll;
            if(i-k>=0) jy=(jy+sum(y,i-k))%mo;
            if(i+k<=m) jy=(jy+sum(y,m)-sum(y,i+k-1)+mo)%mo;
            if(i-k==i+k) jy=(jy-sum(y,i-k)+sum(y,i-k-1)+mo)%mo;
            f[x][i]=(f[x][i]*jy)%mo;
        }
    }
    s[x][0]=0ll;
    fo(i,1,nn+1) s[x][i]=(s[x][i-1]+f[x][i])%mo;
}
int main()
{
    freopen("label.in","r",stdin);freopen("label.out","w",stdout);
    int ac;
    for(scanf("%d",&ac);ac;ac--)
    {
        scanf("%d%lld%d",&n,&m,&k);tot=0;cl(last);
        if(m<=10000) nn=m;else nn=(n-1)*k+1;
        fo(i,1,n-1)
        {
            int x,y;scanf("%d%d",&x,&y);putin(x,y);putin(y,x);
        }
        fo(i,0,n) fo(j,0,30099) f[i][j]=1ll;
        dg(1,0);
        printf("%lld\n",sum(1,m));
    }
    fclose(stdin);fclose(stdout);
}