清北test2 T3树上dp或递推

思路

1. 要求到一个点距离为奇数的所有节点的距离和及偶数（记为ansodd[i]和anseven[i])
2. 那么我们直接求出所有的ans
3. 如何？？
4. 先dfs o(n)求出节点1的ansodd和anseven。(同时维护一个sizodd及sizeven表示i点子树内到他距离为奇数的个数)。
5. 然后我们发现不需要对每个点都o(n)跑一边DFS，我们发现一个节点的ans可以有他的父亲转移过来。
6. 通过讨论边的奇偶，我们分两种情况转移
7. 转移

	if(w%2)
	{
		ansodd[v]=anseven[u]+(cnteven-sizodd[v])*w-sizodd[v]*w;
		anseven[v]=ansodd[u]+(cntodd-sizeven[v])*w-sizeven[v]*w;
		dfs1(v,u,cnteven,cntodd);
	}
	else
	{
		ansodd[v]=ansodd[u]+(cntodd-sizodd[v])*w-sizodd[v]*w;
		anseven[v]=anseven[u]+(cnteven-sizeven[v])*w-sizeven[v]*w;
		dfs1(v,u,cntodd,cnteven);
	}

8. 输出ans o(1)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<cmath>
#include<cstdlib>
#include<ctime>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int N=1e5+100;
const int M=2e5+100;
inline int read(){
	char ch=' ';int f=1;int x=0;
	while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
	return x*f;
}
struct node
{
	int v,nxt;
	int w;
}edge[M];
int head[N],cnt;
void add(int u,int v,int w)
{
	cnt++;
	edge[cnt].v=v;
	edge[cnt].w=w;
	edge[cnt].nxt=head[u];
	head[u]=cnt;
}
int ansodd[N],anseven[N];
int sizodd[N],sizeven[N];
int dis[N];
void dfs(int u,int fa,int sum)
{
	dis[u]=sum;
	sizeven[u]=1;
	sizodd[u]=0;
	for(int i=head[u];i;i=edge[i].nxt)
	{
		int v=edge[i].v;
		int w=edge[i].w;
		if(v!=fa)
		{
			dfs(v,u,sum+w);
			if(w%2)
			{
				sizodd[u]+=sizeven[v];
				sizeven[u]+=sizodd[v];
			}
			else
			{
				sizodd[u]+=sizodd[v];
				sizeven[u]+=sizeven[v];
			}
		}
	}
}
void dfs1(int u,int fa,int cntodd,int cnteven)
{
	for(int i=head[u];i;i=edge[i].nxt)
	{
		int v=edge[i].v;
		int w=edge[i].w;
		if(v!=fa)
		{
			if(w%2)
			{
				ansodd[v]=anseven[u]+(cnteven-sizodd[v])*w-sizodd[v]*w;
				anseven[v]=ansodd[u]+(cntodd-sizeven[v])*w-sizeven[v]*w;
				dfs1(v,u,cnteven,cntodd);
			}
			else
			{
				ansodd[v]=ansodd[u]+(cntodd-sizodd[v])*w-sizodd[v]*w;
				anseven[v]=anseven[u]+(cnteven-sizeven[v])*w-sizeven[v]*w;
				dfs1(v,u,cntodd,cnteven);
			}
		}
	}
}
int main()
{
	int n,q;
	n=read();q=read();
	int i,j;
	for(i=1;i<=n-1;i++)
	{
		int u,v,w;
		u=read();
		v=read();
		w=read();
		add(u,v,w);
		add(v,u,w);
	}
	dfs(1,0,0);
	for(i=1;i<=n;i++)
	{
		if(dis[i]%2)
		ansodd[1]+=dis[i];
		else
		anseven[1]+=dis[i];
	}
	dfs1(1,0,sizodd[1],sizeven[1]);
	for(i=1;i<=q;i++)
	{
		int x=read();
		printf("%d %d\n",ansodd[x],anseven[x]);
	}
	return 0;
}