2019HNCPC I 2019 （树形dp）

题意：树上路径 为 2019 的 点对个数
（比赛时，没写出来，很蠢）
代码：

#include <bits/stdc++.h>
using namespace std;
const int N=2e4+9;
int n,x,y,cnt,w,head[N],ans=0,dp[N][2020];
bool vis[N];
struct Edeg{
	int to,val,next;
}e[N*2];
void add(int x,int y,int val){
	e[cnt]={y,val,head[x]};
	head[x]=cnt++;
}
void dfs(int nowp,int fa){
	dp[nowp][0]=1;
	vis[nowp]=true;
	for(int i=head[nowp];i!=-1;i=e[i].next){
		int to=e[i].to,w=e[i].val;
		if(to!=fa&&vis[e[i].to]==false) dfs(to,nowp);
		else continue;
		 for(int j=0;j<2019;j++) {
            ans+=dp[nowp][j]*dp[to][(2019-j-w+2019)%2019];
        }
        for(int j=0;j<2019;j++) {
            dp[nowp][(j+w)%2019]+=dp[to][j];
        }
	}
	return;
}
int main(){
	while(~scanf("%d",&n)){
		ans=cnt=0;
		for(int i=1;i<=n;i++) {
			head[i]=-1;vis[i]=false;
			for(int j=0;j<2019;j++){
				dp[i][j]=0;
			}
		}
		for(int i=1;i<=n-1;i++){
			scanf("%d %d %d",&x,&y,&w);
			add(x,y,w);
			add(y,x,w);
		}
		dfs(1,0);
		printf("%d\n",ans);
	}
	return 0;
}