https://ac.nowcoder.com/acm/contest/8563/B
思路:画一颗树出来,从子树开始,如果该子树可以是整除块,那么和父亲节点的连线就完全没有必要连。反之为了保证题目的连通块整除要求,是一定要连的。
所以得出原则,能除就不连,不能除就连。在过程中把不能连的点权累加到父亲节点上去。
#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=1e6+100;
typedef long long LL;
typedef pair<LL,LL>P;
LL n,k;
LL w[maxn];///代表点权
vector<P>g[maxn];///first代表节点,second代表边权
LL ans=0;
void dfs(LL u,LL fa)
{
for(LL i=0;i<g[u].size();i++)
{
P to=g[u][i];
LL v=to.first;
if(v==fa) continue;
dfs(v,u);
if(w[v]%k==0) continue;
w[u]+=w[v];
ans+=to.second;
}
}
int main(void)
{
cin.tie(0);std::ios::sync_with_stdio(false);
cin>>n>>k;
for(LL i=1;i<=n;i++) cin>>w[i];
for(LL i=1;i<n;i++){
LL u,v,cost;cin>>u>>v>>cost;
g[u].push_back({v,cost});
g[v].push_back({u,cost});
}
dfs(1,-1);
cout<<ans<<endl;
return 0;
}