【GDSOI2017】 逃亡

Description

一棵树有N个点，每个点有两个属性a,b，a已知，b未知

一个点u，会对它子树内的任意一点v发动a[u]次战争，当且仅当b[u]>b[v]

0<=bi<=m，对于j=0~k，求有多少种方案，使得发动的战争次数恰好为j

0 < n ≤ 14, k ≤ 20, 0 ≤ m ≤ 100000000, 0 ≤ ai, bi ≤ m

Analysis

看到这种数据范围很容易想到状压
首先就可以发现，m那么大是没有用的，答案乘个组合数就好
设f[s][i][j]表示n个点涂色状态为s，造成了i次战争，颜色数枚举到j的方案数
我一开始的想法便是枚举一个点染色，颜色要么是j要么是j+1，但是很显然会算重
后来听说了40分的方法可以直接每一个颜色涂哪些点，枚举涂该颜色的点集
结果优化还是要单点枚举==
为了避免算重，我们发现算重的情况当且仅当存在某个v是u子树的点，u,v颜色相同，v先做，u后做
那么就要保证，当前涂色的点为v，颜色为j，v点子树中不存在颜色为j的点
对于这个，我们只需将点的枚举顺序调整一下，比如按dfs序，拓扑序，深度序等等
当然还要先枚举点，再枚举状态

Summary

此题状压dp模型显然，状态很好想，但是转移及其优化需要一些思维含量
尤其是对于方案数的计算，确定枚举顺序相当重要
细节打错比较难调，建议不放过每一个细节，肉眼查错
出一些极端小数据可能有所帮助

Code

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int N=15,K=23,mo=1e9+7,MX=100000000;
int n,m,k,_2[N],in[N],tp[N],a[N],fa[N];
ll f[16390][K][N];
bool pd[N][N];
queue<int> q;
ll qmi(ll x,int n)
{
    ll t=1;
    for(;n;n>>=1)
    {
        if(n&1) t=t*x%mo;
        x=x*x%mo;
    }
    return t;
}
int main()
{
    freopen("escape.in","r",stdin);
    freopen("escape.out","w",stdout);
    _2[0]=1;
    fo(i,1,14) _2[i]=_2[i-1]*2;
    scanf("%d %d %d",&n,&m,&k);
    fo(i,1,n) scanf("%d",&a[i]);
    fo(i,2,n) scanf("%d",&fa[i]),in[fa[i]]++;
    fo(i,1,n)
        for(int j=i;j;j=fa[j]) pd[j][i]=1;
    fo(i,1,n)
        if(!in[i]) q.push(i);
    while(!q.empty())
    {
        int x=q.front();q.pop();
        tp[++tp[0]]=x;
        if(!--in[fa[x]]) q.push(fa[x]);
    }
    fo(s,1,_2[n]-1) f[s][0][0]=1;
    int mn=min(n-1,m);
    fo(j,1,mn)
        fd(l,n,1)
        {
            int v=tp[l];
            fo(s,0,_2[n]-1) if(!(s&_2[v-1]))
            {
                int num=0;
                fo(u,1,n)
                    if((s&_2[u-1]) && pd[v][u]) num++;
                int s1=s|_2[v-1],dt=a[v]*num;
                fo(i,0,k-dt)
                    f[s1][i+dt][j]=(f[s1][i+dt][j]+f[s][i][j-1]+f[s][i][j])%mo;
            }
        }
    ll x=1;
    fo(l,0,k)
    {
        ll ans=0;
        fo(j,0,mn)
        {
            ll tot=1;
            if(j<m)
            {
                fo(i,m-j+1,m+1) tot=tot*i%mo;
                fo(i,1,j+1) tot=tot*qmi(i,mo-2)%mo;
            }
            ans=(ans+tot*f[_2[n]-1][l][j]%mo)%mo;
        }
        printf("%lld\n",ans);
    }
    return 0;
}