10501 2024. 04. 03 15:52:19 111 A Day in Olbia cpp17 Elfogadva 100/100 2.128s 229068 KiB 
#include<bits/stdc++.h>
using namespace std;

#define int long long

signed main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int N;
	cin>>N;
	vector<vector<int>>g(N);
	for(int i=0;i<N-1;i++){
		int a,b;
		cin>>a>>b;
		g[a].push_back(b);
		g[b].push_back(a);
	}
	int ans=0;
	auto dfs=[&](auto self,int i,int p,int d)->map<pair<int,int>,int>{
		map<pair<int,int>,int>s;
		for(int m=1;i*m<=N&&m<=N;m++){
			s[{m,i*m-d}]++;
		}
		for(int j:g[i]){
			if(j==p){
				continue;
			}
			auto z=self(self,j,i,d+1);
			if(s.size()<z.size()){
				swap(s,z);
			}
			for(auto[mx,y]:z){
				auto[m,x]=mx;
				auto t=s.find({m,-x-d*2});
				if(t!=s.end()){
					ans+=t->second*y;
				}
			}
			for(auto[mx,y]:z){
				s[mx]+=y;
			}
		}
		return s;
	};
	dfs(dfs,0,-1,0);
	cout<<ans<<'\n';
	return 0;
}
Részfeladat Összpont Teszt Verdikt Idő Memória
subtask1 0/0
1 Elfogadva 3ms 1828 KiB
2 Elfogadva 3ms 2056 KiB
subtask2 16/16
3 Elfogadva 3ms 2464 KiB
4 Elfogadva 3ms 2740 KiB
5 Elfogadva 3ms 2920 KiB
6 Elfogadva 3ms 3148 KiB
subtask3 16/16
7 Elfogadva 8ms 4292 KiB
8 Elfogadva 8ms 4072 KiB
9 Elfogadva 6ms 4820 KiB
10 Elfogadva 6ms 4640 KiB
11 Elfogadva 6ms 4776 KiB
12 Elfogadva 6ms 4868 KiB
13 Elfogadva 7ms 5112 KiB
14 Elfogadva 6ms 5056 KiB
15 Elfogadva 6ms 5392 KiB
subtask4 16/16
16 Elfogadva 1.725s 183272 KiB
17 Elfogadva 1.764s 181260 KiB
subtask5 52/52
18 Elfogadva 2.115s 185500 KiB
19 Elfogadva 2.128s 201352 KiB
20 Elfogadva 1.223s 223016 KiB
21 Elfogadva 1.213s 208328 KiB
22 Elfogadva 1.217s 214644 KiB
23 Elfogadva 1.195s 210484 KiB
24 Elfogadva 1.192s 218408 KiB
25 Elfogadva 1.251s 229068 KiB
26 Elfogadva 1.231s 226204 KiB
27 Elfogadva 1.052s 224148 KiB
28 Elfogadva 982ms 223592 KiB
29 Elfogadva 931ms 223912 KiB
30 Elfogadva 1.078s 225024 KiB
31 Elfogadva 1.088s 227820 KiB