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

题目描述

输入

输出

样例输入

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

样例输出

4
2
12

数据范围

样例解释

解法

设f[i][j]为在第i个点填了j的合法方案。
则f[i][j]=∏(f[son(i)][l])(l∈[1,j−k]∪[j+k,m])。
时间复杂度为O(nm)。
观察到f值呈对称状，且中间一段的值是相同的。
利用这个性质优化动态规划。

代码

#include<iostream>#include<stdio.h>#include<math.h>#include<string.h>#include<algorithm>#define ll long long#define ln(x,y) ll(log(x)/log(y))#define sqr(x) ((x)*(x))using namespace std;const char* fin="label.in";const char* fout="label.out";const ll inf=0x7fffffff;const ll maxn=107,maxm=11007,mo=1000000007;ll t,n,m,delta,i,j,k,tot,ans,tmd;ll fi[maxn],la[maxn*2],ne[maxn*2];ll f[maxn][maxm],g[maxn][maxm],h[maxn][maxm];void add_line(ll a,ll b){    tot++;    ne[tot]=fi[a];    la[tot]=b;    fi[a]=tot;}void dfs(ll v,ll from){    ll i,j,k,tmp,pre=-1,tmb=0,o;    for (i=1;i<maxm;i++) f[v][i]=1;    for (k=fi[v];k;k=ne[k])        if (la[k]!=from){            dfs(la[k],v);            for (i=1;i<=tmd;i++){                tmp=0;                if (delta==0){                    tmp=(tmp+g[la[k]][i]+h[la[k]][i]+mo-f[la[k]][i])%mo;                }else{                    j=i-delta;                    if (j>0) tmp=(tmp+g[la[k]][j])%mo;                    j=i+delta;                    if (j<=m) tmp=(tmp+h[la[k]][j])%mo;                }                f[v][i]=(f[v][i]*tmp)%mo;            }        }    for (i=2,j=tmd;i<=j;i++) if (f[v][i]==f[v][i-1]) {        pre=i-2;        break;    }else tmb=(tmb+f[v][i-1])%mo;    if (pre==-1) pre=tmd-1;//,tmb=(tmb+f[v][pre])%mo;    for (i=1;i<=pre;i++) g[v][i]=(g[v][i-1]+f[v][i])%mo;    for (j=min(m-pre+1,maxm);i<j;i++) g[v][i]=(g[v][i-1]+f[v][pre+1])%mo;    for (j=min(m,maxm-1);i<=j;i++) g[v][i]=(g[v][i-1]+f[v][m-i+1])%mo;    h[v][1]=(tmb*2+(m-2*pre+mo)%mo*f[v][pre+1])%mo;    for (i=2;i<=pre+1;i++) h[v][i]=(h[v][i-1]-f[v][i-1]+mo)%mo;    for (j=min(m-pre+1,maxm-1);i<=j && i<=m;i++) h[v][i]=(h[v][i-1]+mo-f[v][pre+1])%mo;    for (j=min(m,maxm-1);i<=j;i++) h[v][i]=(h[v][i-1]+mo-f[v][m-i+2])%mo;}int main(){    freopen(fin,"r",stdin);    freopen(fout,"w",stdout);    scanf("%d",&t);    for (;t;t--){        scanf("%d%d%d",&n,&m,&delta);        tot=0;        tmd=min((m+1)/2,maxm-delta-5);        memset(fi,0,sizeof(fi));        memset(h,0,sizeof(h));        memset(g,0,sizeof(g));        for (i=1;i<n;i++){            scanf("%d%d",&j,&k);            add_line(j,k);            add_line(k,j);        }        dfs(1,0);        ans=h[1][1];        printf("%lld\n",ans);    }    return 0;}

启发

观察转移方程的性质，譬如对称之类的，可以优化动态规划。