标签：struct 2021.7 19 T4 yyy int 复杂度 dp define
首先这个\(60\)分的树形dp很好打，直接裸的树上背包合并即可。加了一个每次与子树大小取min的剪枝。
时间复杂度是\(O(nk^2)\)
然后写了一发过了。
然后再考虑这个东西的复杂度。
首先考虑产生\(k^2\)贡献的时候，这个显然是只有\(O(\frac{n}{k})\)次的。
然后如果是小于\(k\)合并到大于\(k\)，容易发现对于小于\(k\)子树内的每个节点全程只有一次，所以是每个点均摊\(O(k)\)
如果两颗子树都小于\(k\)那么对于每个这种点的贡献是\(O(siz)\)也就是\(O(k)\)
所以时间复杂度\(O(nk)\)
code:
#include<bits/stdc++.h>
#define I inline
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define abs(x) ((x)>0?(x):-(x))
#define re register
#define ll long long
#define db double
#define N 100000
#define M 200
#define mod 1000000007
#define mod2 39989
#define eps (1e-7)
#define U unsigned int
#define it iterator
#define Gc() getchar()
#define Me(x,y) memset(x,y,sizeof(x))
using namespace std;
int n,m,k,x,y,z,siz[N+5],g[M+5][4],dp[N+5][M+5][4];
struct yyy{int to,z;};
struct ljb{
int head,h[N+5];yyy f[N+5<<1];
I void add(int x,int y){f[++head]=(yyy){y,h[x]};h[x]=head;}
}s;
I void dfs(int x,int last){
int i,j,h;yyy tmp;siz[x]=1;dp[x][1][0]=dp[x][0][2]=1;for(int h=s.h[x];h;h=tmp.z){
tmp=s.f[h];if(tmp.to==last) continue;dfs(tmp.to,x);
for(i=0;i<=k;i++) memcpy(g[i],dp[x][i],sizeof(g[i])),Me(dp[x][i],0);
for(i=min(siz[x],k);~i;i--){
for(j=0;j<=siz[tmp.to]&&j+i<=k;j++){
dp[x][i+j][0]=(dp[x][i+j][0]+1ll*g[i][0]*(dp[tmp.to][j][2]+dp[tmp.to][j][3]))%mod;
dp[x][i+j][1]=(dp[x][i+j][1]+1ll*g[i][1]*(1ll*dp[tmp.to][j][0]+dp[tmp.to][j][1]+dp[tmp.to][j][2]+dp[tmp.to][j][3])+1ll*g[i][0]*(dp[tmp.to][j][1]+dp[tmp.to][j][0]))%mod;
dp[x][i+j][2]=(dp[x][i+j][2]+1ll*g[i][2]*dp[tmp.to][j][3])%mod;
dp[x][i+j][3]=(dp[x][i+j][3]+1ll*g[i][3]*(dp[tmp.to][j][1]+dp[tmp.to][j][3])+1ll*g[i][2]*dp[tmp.to][j][1])%mod;
}//dp[x][i][0]=dp[x][i][2]=0;
}siz[x]+=siz[tmp.to];
}
}
int main(){
freopen("disiti.in","r",stdin);freopen("disiti.out","w",stdout);
re int i;scanf("%d%d",&n,&k);for(i=1;i<n;i++) scanf("%d %d",&x,&y),s.add(x,y),s.add(y,x);dfs(1,0);printf("%d\n",(dp[1][k][1]+dp[1][k][3])%mod);
}
标签：struct,2021.7,19,T4,yyy,int,复杂度,dp,define 来源： https://www.cnblogs.com/275307894a/p/15033075.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。