102022024-03-29 13:32:51111Testvérvárosokcpp17Időlimit túllépés 15/1001.582s43696 KiB
#include <bits/stdc++.h>
using namespace std;

#define int long long

signed main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int N,K;
	cin>>N>>K;
	vector<vector<pair<int,int>>>g(N+1);
	for(int i=0;i<N-1;i++){
		int a,b,d;
		cin>>a>>b>>d;
		g[a].emplace_back(b,d);
		g[b].emplace_back(a,d);
	}
	int ans=0;
	auto dfs=[&](auto self,int i,int p,int x)->multiset<int>{
		multiset<int>s;
		for(auto[j,d]:g[i]){
			if(j==p){
				continue;
			}
			auto z=self(self,j,i,x+d);
			ans+=z.count(x%K);
			if(z.size()>s.size()){
				swap(s,z);
			}
			for(int y:z){
				ans+=s.count(((x*2-y)%K+K)%K);
			}
			s.insert(z.begin(),z.end());
		}
		s.insert(x%K);
		return s;
	};
	dfs(dfs,1,0,0);
	cout<<ans<<'\n';
	return 0;
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
subtask10/0
1Elfogadva3ms1824 KiB
2Elfogadva3ms2072 KiB
subtask215/15
3Elfogadva3ms2284 KiB
4Elfogadva3ms2440 KiB
5Elfogadva3ms2768 KiB
6Elfogadva3ms2768 KiB
7Elfogadva4ms2892 KiB
8Elfogadva3ms2852 KiB
9Elfogadva4ms3140 KiB
subtask30/15
10Elfogadva363ms6768 KiB
11Elfogadva24ms5580 KiB
12Időlimit túllépés1.582s8432 KiB
13Időlimit túllépés1.582s7468 KiB
14Időlimit túllépés1.562s7844 KiB
subtask40/20
15Elfogadva3ms3508 KiB
16Elfogadva3ms4152 KiB
17Elfogadva8ms8216 KiB
18Elfogadva37ms21764 KiB
19Elfogadva82ms29640 KiB
20Elfogadva225ms39440 KiB
21Időlimit túllépés1.577s18580 KiB
22Elfogadva97ms43696 KiB
subtask50/50
23Elfogadva100ms16096 KiB
24Elfogadva50ms10660 KiB
25Elfogadva68ms13364 KiB
26Elfogadva23ms6540 KiB
27Elfogadva50ms10536 KiB
28Elfogadva134ms17032 KiB
29Elfogadva96ms16232 KiB
30Időlimit túllépés1.552s8700 KiB
31Elfogadva108ms17560 KiB