洛谷P4516：[JSOI2018]潜入行动（树形dp）

最新推荐文章于 2022-11-23 21:33:50 发布

Fat_tu

最新推荐文章于 2022-11-23 21:33:50 发布

阅读量473

点赞数 1

分类专栏： dp

本文链接：https://blog.csdn.net/q582116859/article/details/80359244

版权

dp 专栏收录该内容

24 篇文章 1 订阅

订阅专栏

本文介绍了一种使用树形动态规划解决特定问题的方法。通过定义状态f[i][j][0/1][0/1]，表示节点i的子树中选择了j个节点，i节点是否被选择及是否被覆盖的情况。文章提供了详细的实现过程，并通过类似卷积的方式合并子树状态，最终求解整个树的最佳方案。

摘要由CSDN通过智能技术生成

题面

大概就是树形dp
设 $f[i][j][0/1][0/1]$ ，表示 $i$ 的子树里选了 $j$ 个， $i$ 有没有选， $i$ 是否被覆盖
然后类似卷积的合并，就能算出答案

一个显然的性质，节点i的次数界为 $min(size[i],k)$
根据某科学的理论

若合并的两棵树 $size$ 都大于等于k，则合并次数不超过 $n/k$

若只有一棵的 $siz$ 大于等于k，相当于小的那棵树的每个点都用k的时间，使其次数界变为k

若两棵 $siz$ 都小于k，设一棵为 $a$
相当于另一棵树的每个点用 $a$ 的时间使其所在树的次数界扩大a

所以就是 $O(nk)$ 的

#include <iostream>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>

using namespace std;
#define mmst(a, b) memset(a, b, sizeof(a))
#define mmcp(a, b) memcpy(a, b, sizeof(b))

typedef long long LL;

void read(int &hy)
{
    hy=0;
    char cc=getchar();
    while(cc<'0'||cc>'9')
    cc=getchar();
    while(cc>='0'&&cc<='9')
    {
        hy=(hy<<3)+(hy<<1)+cc-'0';
        cc=getchar();
    }
}

const int N=100200,p=1e9+7;

int n,k;
int head[N],nex[N<<1],to[N<<1],cnt;
int siz[N];
int f[N][102][2][2],g[102][2][2];

void add(int u,int v)
{
    to[++cnt]=v;
    nex[cnt]=head[u];
    head[u]=cnt;
}

void B(int &x,int a,int b)
{
    x=(x+(LL)a*b%p)%p;
}

void dfs(int x)
{
    siz[x]=1;
    f[x][0][0][0]=f[x][1][1][0]=1;
    for(int h=head[x];h;h=nex[h])
    if(!siz[to[h]])
    {
        int v=to[h];
        dfs(to[h]);

        for(int i=0;i<=min(siz[x]+siz[v],k);i++)
        for(int a=0;a<=1;a++)
        for(int b=0;b<=1;b++)
        g[i][a][b]=0;

        for(int i=0;i<=min(siz[x],k);i++)
        for(int j=0;j<=min(k-i,siz[v]);j++)
        for(int a=0;a<=1;a++)
        for(int b=0;b<=1;b++)
        for(int c=0;c<=1;c++)
        for(int d=0;d<=1;d++)
        if(a+d>0)
        B(g[i+j][a][(b|c)],f[x][i][a][b],f[v][j][c][d]);

        siz[x]+=siz[to[h]];

        for(int i=0;i<=min(siz[x],k);i++)
        for(int a=0;a<=1;a++)
        for(int b=0;b<=1;b++)
        f[x][i][a][b]=g[i][a][b];
    }
}

int main()
{
    cin>>n>>k;

    for(int i=1;i<n;i++)
    {
        int u,v;
        read(u);
        read(v);
        add(u,v);
        add(v,u);
    }

    dfs(1);

    cout<<(f[1][k][0][1]+f[1][k][1][1])%p<<endl;

    return 0;
}