Codeforces868E Policeman and a Tree -- DP

令 fu,v,s1,s2 表示当前在第 u 个点，将要走到第 v 个点，前面有 s1 个罪犯，后面有 s2 个罪犯，将所有罪犯抓到的最小时间。
那么如果 v 是叶子，下一步可以直接抓到 s1 个罪犯，从 fv,u,s2,0 转移。
否则，考虑在 T 时间内是否可以抓到：
判断是否存在一种方案，将 s1 个罪犯分到 v 的所有子树中，使下一步到每棵子树的最小时间都大于 T 。如果存在，那么在 T 时间内无法抓到。
然后用类似背包的做法。
时间复杂度 O(n5) ，但似乎远小于 n5 。

#include
#include
#include
#include
#include
using namespace std;
#define N 60
#define INF 1e9
vector<int>g[N];
int i,j,k,n,m,p,a[N],c[N][N],f[2][N];
int dp[N][N][N][N];
int x,y,z,Rt,Ans=INF;
inline void Dfs(int x,int y){
    for(int i=0;iif(g[x][i]!=y)Dfs(g[x][i],x),a[x]+=a[g[x][i]];
}
inline int Solve(int x,int y,int s1,int s2){
    if(!s1&&!s2)return 0;
    if(!s1)return INF;
    int& A=dp[x][y][s1][s2];
    if(Areturn A;
    if(g[y].size()==1)return (A=Solve(y,x,s2,0)+c[x][y]);
    for(int i=0;iif(g[y][i]!=x)
    for(int k=0;k<=s1;k++)Solve(y,g[y][i],k,s1+s2-k);
    memset(f[1],0,sizeof(f[1]));
    f[1][0]=INF;
    bool b=0;
    for(int i=0;iint v=g[y][i];
        if(v==x)continue;
        memset(f[b],0,sizeof(f[b]));
        for(int j=0;j<=s1;j++)
        for(int k=0;k+j<=s1;k++)
        f[b][j+k]=max(f[b][j+k],min(f[b^1][j],Solve(y,v,k,s1+s2-k)));
        b^=1;
    }
    return (A=f[b^1][s1]+c[x][y]);
}
int main(){
    scanf("%d",&n);
    for(i=1;iscanf("%d%d%d",&x,&y,&z);
        g[x].push_back(y);g[y].push_back(x);
        c[x][y]=c[y][x]=z;
    }
    scanf("%d%d",&Rt,&m);
    for(i=1;i<=m;i++)scanf("%d",&x),a[x]++;
    Dfs(Rt,0);
    memset(dp,0x3f,sizeof(dp));
    for(i=0;icout<return 0;
}