zoukankan      html  css  js  c++  java
  • BZOJ 5314: [Jsoi2018]潜入行动

    树形DP

    #include<cstdio>
    #include<algorithm>
    using namespace std;
    const int mod=1e9+7;
    int cnt,n,k,F[100005][105][2][2],last[100005],sz[100005],G[105][2][2];
    struct node{
    	int to,next;
    }e[200005];
    void add(int a,int b){
    	e[++cnt].to=b;
    	e[cnt].next=last[a];
    	last[a]=cnt;
    }
    void dfs(int x,int fa){
    	sz[x]=1;
    	F[x][0][0][0]=1,F[x][1][1][0]=1;
    	for (int i=last[x]; i; i=e[i].next){
    		int V=e[i].to;
    		if (V==fa) continue;
    		dfs(V,x);
    		for (int now=0; now<=min(sz[x],k); now++){
    			G[now][0][0]=F[x][now][0][0],F[x][now][0][0]=0;
    			G[now][0][1]=F[x][now][0][1],F[x][now][0][1]=0;
    			G[now][1][0]=F[x][now][1][0],F[x][now][1][0]=0;
    			G[now][1][1]=F[x][now][1][1],F[x][now][1][1]=0;
    		}
    		for (int pre=0; pre<=min(sz[x],k); pre++)
    			for (int now=0; now<=min(sz[V],k-pre); now++){
    				int To=pre+now;
    				F[x][To][0][0]+=1ll*G[pre][0][0]*F[V][now][0][1]%mod; if (F[x][To][0][0]>mod) F[x][To][0][0]-=mod;
    				F[x][To][0][1]+=1ll*G[pre][0][0]*F[V][now][1][1]%mod; if (F[x][To][0][1]>mod) F[x][To][0][1]-=mod;
    				F[x][To][0][1]+=1ll*G[pre][0][1]*(F[V][now][0][1]+F[V][now][1][1])%mod; if (F[x][To][0][1]>mod) F[x][To][0][1]-=mod;
    				F[x][To][1][0]+=1ll*G[pre][1][0]*(F[V][now][0][0]+F[V][now][0][1])%mod; if (F[x][To][1][0]>mod) F[x][To][1][0]-=mod;
    				F[x][To][1][1]+=1ll*G[pre][1][0]*(F[V][now][1][0]+F[V][now][1][1])%mod; if (F[x][To][1][1]>mod) F[x][To][1][1]-=mod;
    				F[x][To][1][1]+=1ll*G[pre][1][1]*(1ll*F[V][now][0][0]+F[V][now][0][1]+F[V][now][1][0]+F[V][now][1][1])%mod;
    				if (F[x][To][1][1]>mod) F[x][To][1][1]-=mod;
    			}
    		sz[x]+=sz[V];
    	}
    }
    int main(){
    	scanf("%d%d",&n,&k);
    	for (int i=1; i<n; i++){
    		int x,y;
    		scanf("%d%d",&x,&y);
    		add(x,y);
    		add(y,x);
    	}
    	dfs(1,0);
    	printf("%d
    ",(F[1][k][0][1]+F[1][k][1][1])%mod);
    	return 0;
    }
    

      

  • 相关阅读:
    vue项目中引用spreadjs方法
    使用Element的table合并单元格的问题(合并行)
    卸载mysql
    mac 下安装mySQL
    react中界面跳转 A界面跳B界面,返回A界面,A界面状态保持不变 localStorage方法
    react点击弹出带出值和点击跳转页面带出值
    ajax优缺点
    vue项目console.log报错
    性能优化
    对象生命周期
  • 原文地址:https://www.cnblogs.com/silenty/p/9911239.html
Copyright © 2011-2022 走看看