首页 > 其他分享 >【南外】DPS1-A

【南外】DPS1-A

时间:2023-09-15 20:46:42浏览次数:40  
标签:DPS1 tmp int Vi dfs 南外 lst define

/*
树形DP,记 f[i][j] 表示第 i 个节点,其子树中包含它的长度为 j 的链的数量。
对于一个节点 x,其有子节点 V1,V2,...
设当前做到 Vi,则 lst[i][...] 表示做到 Vi-1 时 f[x][...] 的值。
统计答案时,对于新增点 Vi,假设我们已经跟新完了 f(即 f[x][j] 加上 f[Vi][j-1]), 
则答案的增加量为 Σ(1<=j<=k) (f[x][j]-lst[x][j])*lst[x][k-j]
意为(新增加进来的子节点中长度为 j 的链数量)*(原有的长度为 k-j 的链数量) 
然后更新 lst 即可。 
*/ 

//If, one day, I finally manage to make my dreams a reality...
//I wonder, will you still be there by my side?
#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(false)
#define TIE cin.tie(0),cout.tie(0)
#define int long long
#define y1 cyy
#define fi first
#define se second
#define cnt1(x) __builtin_popcount(x)
#define mk make_pair
#define pb push_back
#define pii pair<int,int>
#define ls(x) (x<<1)
#define rs(x) (x<<1|1)
#define lbt(x) (x&(-x))
using namespace std;
int n,k,u,v,f[50005][505],lst[50005][505],ans;
vector<int> a[200005]; 
void dfs(int x,int fa){
	f[x][0]=lst[x][0]=1;
	for(auto tmp:a[x]){
		if(tmp!=fa){
			dfs(tmp,x);
			for(int i=1;i<=k;i++) f[x][i]+=f[tmp][i-1];
			for(int i=k;i>=1;i--){
				ans+=(f[x][i]-lst[x][i])*lst[x][k-i];
			}
			for(int i=1;i<=k;i++) lst[x][i]+=f[tmp][i-1];
		}
	}
}
signed main(){
	IOS;TIE;
	cin>>n>>k;
	for(int i=1;i<n;i++){
		cin>>u>>v;
		a[u].pb(v);
		a[v].pb(u);
	} 
	dfs(1,0);
	cout<<ans<<endl;
	return 0;
}

标签:DPS1,tmp,int,Vi,dfs,南外,lst,define
From: https://www.cnblogs.com/binary1110011/p/17705893.html

相关文章

  • 2023年 8月15日普及组南外集训题解
    A陷阱我们可以从\(l\)枚举到\(d\),再计算是否满足要求,满足要求加入到数组中,输出第一个和最后一个#include<iostream>usingnamespacestd;constintN=1e5+5;intk;intnums[N];intmain(){intl,d,x;cin>>l>>d>>x;for(inti=l;i<=d......
  • 2023年 8月15日普及组南外集训题解
    A查找最大元素扫一遍确定最大值,如果是最大值输出字符和"(max)",不是的话只输出字符#include<iostream>#include<cstring>usingnamespacestd;charmaxx;strings;intmain(){cin>>s;for(inti=0;i<s.size();i++)if(s[i]>=maxx)......
  • 2023年 8月7日普及组南外集训题解
    A国家集训队题解注意数据已经是有序的,我还搞了个排序,我是智障所以只需要将第5个人到第16个人的成绩都预设成300,再把前4个人的成绩都预设成0,再看有没有人能超过第4个人就行了ac代码#include<iostream>usingnamespacestd;constintN=20;inta[N],ans=4;intmain(......