10498 | 2024-04-03 15:29:46 | 111 | A Day in Olbia | cpp17 | Futási hiba 32/100 | 630ms | 4592 KiB |
#include<bits/stdc++.h>
using namespace std;
#define int long long
int get(map<int,int>&m,int x){
auto t=m.find(x);
return t==m.end()?0:t->second;
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(nullptr);
int N;
cin>>N;
if(N>1000)exit(1);
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;
for(int m=1;m<=N;m++){
auto dfs=[&](auto self,int i,int p,int d)->map<int,int>{
map<int,int>s;
for(int j:g[i]){
if(j==p){
continue;
}
auto z=self(self,j,i,d+1);
ans+=get(z,-(i*m-d)-d*2);
if(s.size()<z.size()){
swap(s,z);
}
for(auto[x,y]:z){
ans+=get(s,-x-d*2)*y;
}
for(auto[x,y]:z){
s[x]+=y;
}
}
s[i*m-d]++;
return s;
};
dfs(dfs,8,-1,0);
}
cout<<ans<<'\n';
return 0;
}
Részfeladat | Összpont | Teszt | Verdikt | Idő | Memória | ||
---|---|---|---|---|---|---|---|
subtask1 | 0/0 | ||||||
1 | Futási hiba | 3ms | 1928 KiB | ||||
2 | Futási hiba | 3ms | 2260 KiB | ||||
subtask2 | 16/16 | ||||||
3 | Elfogadva | 4ms | 2464 KiB | ||||
4 | Elfogadva | 4ms | 2732 KiB | ||||
5 | Elfogadva | 4ms | 2996 KiB | ||||
6 | Elfogadva | 4ms | 2916 KiB | ||||
subtask3 | 16/16 | ||||||
7 | Elfogadva | 620ms | 3160 KiB | ||||
8 | Elfogadva | 630ms | 3388 KiB | ||||
9 | Elfogadva | 160ms | 4092 KiB | ||||
10 | Elfogadva | 230ms | 4004 KiB | ||||
11 | Elfogadva | 202ms | 4212 KiB | ||||
12 | Elfogadva | 174ms | 4260 KiB | ||||
13 | Elfogadva | 187ms | 4376 KiB | ||||
14 | Elfogadva | 202ms | 4448 KiB | ||||
15 | Elfogadva | 165ms | 4592 KiB | ||||
subtask4 | 0/16 | ||||||
16 | Futási hiba | 3ms | 3968 KiB | ||||
17 | Futási hiba | 3ms | 3972 KiB | ||||
subtask5 | 0/52 | ||||||
18 | Futási hiba | 2ms | 4036 KiB | ||||
19 | Futási hiba | 3ms | 4256 KiB | ||||
20 | Futási hiba | 2ms | 4264 KiB | ||||
21 | Futási hiba | 3ms | 4336 KiB | ||||
22 | Futási hiba | 3ms | 4352 KiB | ||||
23 | Futási hiba | 3ms | 4352 KiB | ||||
24 | Futási hiba | 3ms | 4368 KiB | ||||
25 | Futási hiba | 2ms | 4468 KiB | ||||
26 | Futási hiba | 3ms | 4468 KiB | ||||
27 | Futási hiba | 3ms | 4460 KiB | ||||
28 | Futási hiba | 2ms | 4356 KiB | ||||
29 | Futási hiba | 2ms | 4356 KiB | ||||
30 | Futási hiba | 2ms | 4368 KiB | ||||
31 | Futási hiba | 3ms | 4484 KiB |