牛客挑战赛45(我是A题)(思维+贪心)

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;
}