2019牛客国庆集训派对day1 -I 2019_2019牛客集训派对-CSDN博客

本文链接：https://blog.csdn.net/a1_s2_c3_/article/details/102239014

本文介绍了一种使用动态规划算法解决无向图中寻找边权为2019倍数路径的问题。核心思路是通过递归深度优先搜索，计算从每个节点出发，到达其子节点的路径数量，这些路径的边权总和模2019等于特定值。最终统计所有路径数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

无向图中边权是2019倍数的路径数；

题目链接

我这里面有一篇是求边权是3的倍数，见另外一篇文章，点这儿（【luogu P2634】聪聪可可）
。

思路：
思路：dp[i][j]表示点 i 的子节点到点 i 的距离mod 2019 = j的路径个数。设一点为u，其子节点为v，u、v间的边权值为w。

转移方程为：dp[u][j]+=dp[v][(j-w+2019)%2019];

而结果ans += dp[u][j] * dp[u][(2019-j-weihgt) % 2019];
就是和为2019. (j+2019)—到u点的距离，

# include<bits/stdc++.h>
using namespace std;
#include<cstring>
#include<algorithm>
#include<vector>
#define ll long long
using namespace std;
vector <int>mp[20005];
vector <int>w[20005];
int dp[20005][2020];
ll ans=0;



void dfs(int fa, int now)
{
	dp[now][0] = 1;
	for(int i = 0; i < mp[now].size(); ++i)
	{
		int next = mp[now][i];
		int weight = w[now][i];
		if(next == fa)	continue;
		dfs(now, next);
		 for(int j = 0; j < 2019; ++j)
		 {
		 	ans += dp[now][j] * dp[next][(2019-weight-j+2019) % 2019];
		 }
		 for(int j = 0; j < 2019; ++j)
		 {
		 	dp[now][j] += dp[next][(j-weight+2019)%2019];
		 }
	}
}

int main()
{
	int n;
	while(~scanf("%d", &n))
	{
		for(int i = 1;  i <= n; ++i)
		{
			mp[i].clear();
			w[i].clear();
			for(int j = 0; j <= 2019; j++)
			{
				dp[i][j] = 0;
			}
		}
		for(int i = 1; i < n; ++i)
		{
			int a, b, c;
			cin>>a>>b>>c;
			mp[a].push_back(b);
			mp[b].push_back(a);
			w[a].push_back(c);
			w[b].push_back(c);
		}
		ans = 0;
		dfs(0,1);
		cout<<ans<<endl;
	}
	return 0;
 }