<JZOJ5913>林下风气

快乐dp

反正考场写挂

#include<cstdio>
#include<cstring>
#include<cctype>
#include<iostream>
#define MOD 19260817
#define LL long long
template <class T>inline void read(T &X)
{
    X=;int W=;char ch=;
    while(!isdigit(ch))W|=ch=='-',ch=getchar();
    while(isdigit(ch))X=(X<<)+(X<<)+(ch^),ch=getchar();
    X=W?-X:X;return;
}
int a[],n,k,head[],f[][][];//j最小值k最大值
int cnt=,mn,mx,ans;
bool boo[];
struct node{int to,next;}edge[];
void add(int u,int v)
{
    edge[++cnt].to=v;
    edge[cnt].next=head[u];
    head[u]=cnt;
}
void dfs(int x,int fa)
{
    if(mn==mx && a[x]==mn) f[x][][]=; else
        if(a[x]==mx && a[x]==mn) f[x][][]=f[x][][]=f[x][][]=; else
            if(a[x]==mx) f[x][][]=; else
                if(a[x]==mn) f[x][][]=; else f[x][][]=;
    for(int i=head[x];i;i=edge[i].next)
    {
        int to=edge[i].to;
        if(to!=fa)
        {
            dfs(to,x);
            f[x][][]=f[x][][]* ( (LL)f[to][][] +f[to][][]+f[to][][]+f[to][][]+ )%MOD;
            f[x][][]=(f[x][][]+ (LL)f[x][][] * (f[to][][]+f[to][][]) )%MOD;
            f[x][][]=(f[x][][]+ (LL)f[x][][] * (f[to][][]+f[to][][]) )%MOD;
            f[x][][]=(f[x][][]+ (LL)f[x][][] *f[to][][])%MOD;

            f[x][][]=(LL)f[x][][] * (f[to][][]+f[to][][]+)%MOD;
            f[x][][]=(f[x][][]+ (LL)f[x][][] *f[to][][])%MOD;

            f[x][][]=(LL)f[x][][] * (f[to][][]+f[to][][]+)%MOD;
            f[x][][]=(f[x][][]+ (LL)f[x][][] *f[to][][])%MOD;

            f[x][][]=(LL)f[x][][] *(f[to][][]+)%MOD;
        }
    }
    if(a[x]<mn || a[x]>mx) f[x][][]=f[x][][]=f[x][][]=f[x][][]=;
}
int main()
{
//    freopen("lkf.in","r",stdin);
//    freopen("lkf.out","w",stdout);
    read(n),read(k);
    for(int i=;i<=n;i++)
        read(a[i]),boo[a[i]]=true;
    for(int i=;i<n;i++)
    {
        int x,y ;
        read(x),read(y);
        add(x,y),add(y,x);
    }
    for(int i=;i+k<=n;i++)
        if(boo[i] && boo[i+k])//差为k a[i]都小于n！
        {
            mn=i,mx=i+k;//差为k 设为min&max
            memset(f,,sizeof(f));
            dfs(,);
            for(int j=;j<=n;j++) ans=(ans+f[j][][])%MOD;
        }
    printf("%d",ans);
    return ;
}