湖南省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 ,我们需要减去所有 的(i,i)一共n个,然后再除以二就是最后的结果。

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

AC代码(比赛AC,没去中南OJ提交,还没有。。后续测试)

#include
#include
#include
#include
#include
#include
#include
#include
#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]

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

你可能感兴趣的:(算法学习)