I-2019
题意:求从u到v所经过的权值相加%2019==0的对数,要求u>v。
题解:首先你要知道怎么求固定根,用树形dp写出,可以仿照这类题的解法,设dp[i][j]代表节点i权值为j%2019的点的个数,然后换根dp。这个换根dp跟我之前写的不完全一样,建议看代码学习一下。
#include<bits/stdc++.h>
#define pi pair<int,int>
#define mk make_pair
#define ll long long
using namespace std;
const int maxn = 2e4+10,mod = 2019;
vector<pi>G[maxn];
int dp[maxn][2020];
ll ans = 0;
void dfs(int u,int fa)
{
for(auto tmp :G[u])
{
int v = tmp.first , w = tmp.second;
if(v==fa) continue;
dfs(v,u);
dp[u][w]++;
for(int i=0;i<2019;i++)
dp[u][(i+w)%mod] += dp[v][i];
}
}
void dfs2(int u,int fa){
for(auto tmp :G[u])
{
int v = tmp.first , w = tmp.second;
if(v==fa) continue;
int d1[2019],d2[2019];
for(int i=0;i<2019;i++)
d1[i] = dp[u][i],d2[i]=dp[v][i];
for(int i=0;i<2019;i++)
dp[u][(i+w)%mod] -= dp[v][i];
dp[u][w]--;
dp[v][w]++;
for(int i=0;i<2019;i++)
dp[v][(i+w)%mod] += dp[u][i];
dfs2(v,u);
ans += dp[v][0];
for(int i=0;i<2019;i++)
dp[u][i] = d1[i] , dp[v][i] = d2[i];
}
}
int main()
{
int n;
while(~scanf("%d",&n))
{
int u,v,w;
for(int i=1;i<=n;i++)G[i].clear();
for(int i=1;i<=n;i++)
for(int j=0;j<=2019;j++)
dp[i][j] = 0;
ans = 0;
for(int i=1;i<n;i++)
{
scanf("%d%d%d",&u,&v,&w);
G[u].push_back(mk(v,w));
G[v].push_back(mk(u,w));
}
dfs(1,0);
ans = dp[1][0];
dfs2(1,0);
printf("%lld\n",ans/2);
}
}