湖南省2019第十五届程序设计大赛I题：2019

I. 2019

Bobo 有一颗 n 个点的树，点的编号是 1,2,...,n. 树有 (n−1) 条边，第 i 条边的端点是 ai 和 bi，权值是 ci. 求满足 u < v 的 (u,v) 数量，满足点 u 到点 v 路径上的权值和是 2019 的倍数。
输入格式
输入文件包含多组数据，请处理到文件结束。 每组数据的第一行包含一个整数 n. 接下来 (n−1) 行，其中第 i 行包含三个整数 ai, bi 和 ci. • n ≤2×104 • 1≤ ai,bi ≤ n • 0≤ ci < 2019 • n 的总和不超过 105.
输出格式
对于每组数据，输出一个整数，表示所求的值。
样例输入
4 1 2 1 1 3 2018 1 4 1 4 1 2 0 1 3 0 1 4 0 3 1 2 1 2 3 1
样例输出
2 6 0

一看题目求树上路径问题，点分治啊！刚好前段时间写过[国家集训队]聪聪可可，我还是当目模板写的，23333.不过需要注意的是，聪聪可可这题计算的路径多，题目给出的u<v限制 ，我们需要减去所有 的（i,i）一共n个，然后再除以二就是最后的结果。

因为大部分是模板所以没怎么打注释，不懂的先写聪聪可可 然后就懂了，点分治模板题。

AC代码（比赛AC，没去中南OJ提交，还没有。。后续测试）

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<queue>
#include<stack>
#include<map>
#include<algorithm>
#define Max 20010
#define inf 0x3f3f3f3f
#define min(a,b) a>b?b:a;
#define mod 2019
using namespace std;
int max(int x,int y)
{
	if(x>y) return x;
	return y;
}
struct  Node{
	int next;
	int to;
	int v;
}edge[Max<<1];
int num_edge,head[Max];
int maxp[Max],size[Max];
int dis[Max],rem[Max];
int vis[Max];
int d[Max];
int t[2050];
int sum,rt;
int n,ans;
void add_edge(int x,int y,int v) //树的双边
{
	edge[++num_edge].next=head[x],head[x]=num_edge;
	edge[num_edge].to=y,edge[num_edge].v=v;
	edge[++num_edge].next=head[y],head[y]=num_edge;
	edge[num_edge].to=x;edge[num_edge].v=v;
}
void getrt(int u,int pa)  //找重心
{
	size[u]=1;maxp[u]=0;
	for(int i=head[u];i;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v==pa||vis[v]) continue;
		getrt(v,u);
		size[u]+=size[v];
		maxp[u]=max(maxp[u],size[v]);
	}
	maxp[u]=max(maxp[u],sum-size[u]);
	if(maxp[u]<maxp[rt]) rt=u;
}
void getdeep(int u,int fa) //计算所有儿子
{
	t[d[u]]++;
	for(int i=head[u];i;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v==fa||vis[v]) continue;
		d[v]=(d[u]+edge[i].v)%mod;//下个点的距离
		getdeep(v,u);
	}

}
int calc(int u,int v)
{
	memset(t,0,sizeof(t));
	d[u]=v;
	getdeep(u,0);//计算完该子树
	int ret=0;
	for(int i=1;i<=1009;i++)
	{
		ret+=t[i]*t[2019-i]*2;
	}
	ret+=t[0]*t[0];
	return ret;//统计
}
void solve(int u)
{
	ans+=calc(u,0);
	vis[u]=1;
	for(int i=head[u];i;i=edge[i].next)
	{
		int v=edge[i].to;
		if(vis[v]) continue;
		ans-=calc(v,edge[i].v);
		maxp[rt=0]=inf;
		sum=size[v];
		getrt(v,0);
		solve(rt);
	}
}
void init()
{
	num_edge=0;ans=0;
	memset(head,0,sizeof(head));
	memset(vis,0,sizeof(vis));
}
int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		init();
		int x,y,w;
		for(int i=1;i<n;i++)
		{
			scanf("%d%d%d",&x,&y,&w);
			w%=mod;
			add_edge(x,y,w);
		}
		rt=0;
		maxp[rt]=n;
		sum=n;
		getrt(1,0);
		solve(rt);
		printf("%d\n",(ans-n)/2);
	}
	
	return 0;
}

算是运气好吧，感觉还有一些简单题没来得及看，还是太菜了。