【树上点对问题】Tree Problem

Problem - D - Codeforces

题意：

思路：

一个很裸的树形DP

对于树上的一对点，我们往往考虑更换枚举对象

如果关注的是路径经过边，那么就考虑一条边两端的连通分量

如果关注的是路径经过点，那么分成两部分考虑

一部分是这个点的子树之间的贡献

另一部分是子树整体和上子树的贡献

更推广一步总结：考虑子树之间的贡献时，可以考虑像树上背包这样的遍历方式

Code：

#include <bits/stdc++.h>

#define int long long

using namespace std;

const int mxn=1e5+10;
const int mxv=1e5+10;
const int mod=1e9+7;

vector<int> G[mxn];

int N,Q,u,v,x;
int sz[mxn],dp[mxn]; 

void dfs(int u,int fa){
	sz[u]=1;
	for(auto v:G[u]){
		if(v==fa) continue;
		dfs(v,u);
		dp[u]+=sz[u]*sz[v];
		sz[u]+=sz[v];
	}
	dp[u]+=sz[u]*(N-sz[u]);
}
void solve(){
	cin>>N;
	for(int i=1;i<=N-1;i++){
		cin>>u>>v;
		G[u].push_back(v);
		G[v].push_back(u);
	}
	dfs(1,0);
	cin>>Q;
	while(Q--){
		cin>>x;
		cout<<dp[x]<<'\n'; 
	}
}
signed main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int __=1;//cin>>__;
	while(__--)solve();return 0;
}