oj.uz

Submission #913875

# Submission time Handle Problem Language Result Execution time Memory
913875 2024-01-20T11:27:09 Z MilosMilutinovic Hard route (IZhO17_road) C++14
52 / 100
 2000 ms 172484 KB 
#include<bits/stdc++.h>
 
#define pb push_back
#define fi first
#define se second
#define mp make_pair
 
using namespace std;
 
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
typedef long double ld;
 
template <typename T> bool chkmin(T &x,T y){return x>y?x=y,1:0;}
template <typename T> bool chkmax(T &x,T y){return x<y?x=y,1:0;}
 
ll readint(){
	ll x=0,f=1; char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}

#define int long long

int n,tot;
int v[1000005],nxt[1000005],h[1000005],mx[500005],cnt[500005];
ll res=0,c=1;
vector<pii> vals[500005];

void addedge(int x,int y){
	v[++tot]=y; nxt[tot]=h[x]; h[x]=tot;
	v[++tot]=x; nxt[tot]=h[y]; h[y]=tot;
}

void dfs(int x,int px,int d){
	mx[x]=d;
	cnt[x]=1;
	for(int p=h[x];p;p=nxt[p]){
		if(v[p]==px) continue;
		dfs(v[p],x,d+1);
		if(mx[v[p]]==mx[x]) cnt[x]+=cnt[v[p]];
		else if(mx[x]<mx[v[p]]) cnt[x]=cnt[v[p]];
		mx[x]=max(mx[x],mx[v[p]]);
	}
}

void reroot(int x,int px,int d){
	vector<pii> f;
	for(int p=h[x];p;p=nxt[p]){
		if(v[p]!=px) mx[v[p]]-=d;
		f.pb(mp(mx[v[p]],cnt[v[p]]));
	}
	sort(f.rbegin(),f.rend());
	vals[x]=f;
	if((int)f.size()>=3){
		ll cur=0,cc=0;
		cur=f[0].fi*1ll*(f[1].fi+f[2].fi);
		ll s=0;
		for(int j=0;j<f.size();j++){
			if(f[j].fi==f[2].fi) cc+=s*f[2].se;
			if(f[j].fi==f[1].fi) s+=f[j].se;
		}
		if(res<cur) res=cur,c=cc;
		else if(res==cur) c+=cc;
	}
	int mx_bef=mx[x],cnt_bef=cnt[x];
	multiset<int> st;
	map<int,int> c;
	for(int p=h[x];p;p=nxt[p]){
		st.insert(mx[v[p]]);
		c[mx[v[p]]]+=cnt[v[p]];
	}
	st.insert(0);
	c[0]+=1;
	for(int p=h[x];p;p=nxt[p]){
		if(v[p]==px) continue;
		st.erase(st.find(mx[v[p]]));
		c[mx[v[p]]]-=cnt[v[p]];
		mx[x]=*prev(st.end())+1;
		cnt[x]=c[mx[x]-1];
		reroot(v[p],x,d+1);
		st.emplace(mx[v[p]]);
		c[mx[v[p]]]+=cnt[v[p]];
	}
	mx[x]=mx_bef;
	cnt[x]=cnt_bef;
}

signed main(){
	n=readint();
	for(int i=1;i<n;i++) addedge(readint(),readint());
	dfs(1,1,0);
	reroot(1,1,0);
	for(int i=1;i<=n;i++){
		dfs(i,i,0);
		vector<pii> f;
		for(int p=h[i];p;p=nxt[p]){
			f.pb(mp(mx[v[p]],cnt[v[p]]));
		}
		sort(f.rbegin(),f.rend());
		assert(vals[i]==f);
	}
	printf("%lld %lld\n",res,c);
	return 0;
}

Compilation message

road.cpp: In function 'void reroot(long long int, long long int, long long int)':
road.cpp:62:16: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   62 |   for(int j=0;j<f.size();j++){
      |               ~^~~~~~~~~

Subtask #1
19.0 / 19.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 19036 KB Output is correct
2 Correct 5 ms 19036 KB Output is correct
3 Correct 5 ms 19036 KB Output is correct
4 Correct 5 ms 19032 KB Output is correct
5 Correct 4 ms 19036 KB Output is correct
6 Correct 5 ms 19036 KB Output is correct
7 Correct 7 ms 19036 KB Output is correct
8 Correct 5 ms 19036 KB Output is correct
9 Correct 5 ms 19036 KB Output is correct
10 Correct 6 ms 19032 KB Output is correct
11 Correct 6 ms 19036 KB Output is correct
12 Correct 5 ms 19036 KB Output is correct
13 Correct 5 ms 19052 KB Output is correct
14 Correct 5 ms 19036 KB Output is correct
15 Correct 5 ms 19036 KB Output is correct
16 Correct 5 ms 19036 KB Output is correct
17 Correct 5 ms 19036 KB Output is correct
18 Correct 6 ms 19036 KB Output is correct
19 Correct 5 ms 19036 KB Output is correct
20 Correct 5 ms 19036 KB Output is correct
21 Correct 5 ms 19036 KB Output is correct
22 Correct 5 ms 19036 KB Output is correct
23 Correct 5 ms 19036 KB Output is correct
24 Correct 5 ms 18856 KB Output is correct

Subtask #2
33.0 / 33.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 19036 KB Output is correct
2 Correct 5 ms 19036 KB Output is correct
3 Correct 5 ms 19036 KB Output is correct
4 Correct 5 ms 19032 KB Output is correct
5 Correct 4 ms 19036 KB Output is correct
6 Correct 5 ms 19036 KB Output is correct
7 Correct 7 ms 19036 KB Output is correct
8 Correct 5 ms 19036 KB Output is correct
9 Correct 5 ms 19036 KB Output is correct
10 Correct 6 ms 19032 KB Output is correct
11 Correct 6 ms 19036 KB Output is correct
12 Correct 5 ms 19036 KB Output is correct
13 Correct 5 ms 19052 KB Output is correct
14 Correct 5 ms 19036 KB Output is correct
15 Correct 5 ms 19036 KB Output is correct
16 Correct 5 ms 19036 KB Output is correct
17 Correct 5 ms 19036 KB Output is correct
18 Correct 6 ms 19036 KB Output is correct
19 Correct 5 ms 19036 KB Output is correct
20 Correct 5 ms 19036 KB Output is correct
21 Correct 5 ms 19036 KB Output is correct
22 Correct 5 ms 19036 KB Output is correct
23 Correct 5 ms 19036 KB Output is correct
24 Correct 5 ms 18856 KB Output is correct
25 Correct 273 ms 20132 KB Output is correct
26 Correct 263 ms 20544 KB Output is correct
27 Correct 251 ms 20572 KB Output is correct
28 Correct 254 ms 20824 KB Output is correct
29 Correct 216 ms 20648 KB Output is correct
30 Correct 215 ms 20900 KB Output is correct
31 Correct 214 ms 20888 KB Output is correct
32 Correct 221 ms 20472 KB Output is correct
33 Correct 271 ms 20568 KB Output is correct
34 Correct 304 ms 20824 KB Output is correct
35 Correct 277 ms 20832 KB Output is correct
36 Correct 298 ms 20824 KB Output is correct
37 Correct 472 ms 21080 KB Output is correct
38 Correct 497 ms 22360 KB Output is correct
39 Correct 439 ms 20056 KB Output is correct
40 Correct 453 ms 19548 KB Output is correct
41 Correct 415 ms 19516 KB Output is correct
42 Correct 407 ms 19332 KB Output is correct
43 Correct 398 ms 19292 KB Output is correct
44 Correct 386 ms 19288 KB Output is correct
45 Correct 311 ms 19292 KB Output is correct
46 Correct 175 ms 19292 KB Output is correct
47 Correct 181 ms 19292 KB Output is correct
48 Correct 148 ms 19572 KB Output is correct

Subtask #3
0 / 48.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 19036 KB Output is correct
2 Correct 5 ms 19036 KB Output is correct
3 Correct 5 ms 19036 KB Output is correct
4 Correct 5 ms 19032 KB Output is correct
5 Correct 4 ms 19036 KB Output is correct
6 Correct 5 ms 19036 KB Output is correct
7 Correct 7 ms 19036 KB Output is correct
8 Correct 5 ms 19036 KB Output is correct
9 Correct 5 ms 19036 KB Output is correct
10 Correct 6 ms 19032 KB Output is correct
11 Correct 6 ms 19036 KB Output is correct
12 Correct 5 ms 19036 KB Output is correct
13 Correct 5 ms 19052 KB Output is correct
14 Correct 5 ms 19036 KB Output is correct
15 Correct 5 ms 19036 KB Output is correct
16 Correct 5 ms 19036 KB Output is correct
17 Correct 5 ms 19036 KB Output is correct
18 Correct 6 ms 19036 KB Output is correct
19 Correct 5 ms 19036 KB Output is correct
20 Correct 5 ms 19036 KB Output is correct
21 Correct 5 ms 19036 KB Output is correct
22 Correct 5 ms 19036 KB Output is correct
23 Correct 5 ms 19036 KB Output is correct
24 Correct 5 ms 18856 KB Output is correct
25 Correct 273 ms 20132 KB Output is correct
26 Correct 263 ms 20544 KB Output is correct
27 Correct 251 ms 20572 KB Output is correct
28 Correct 254 ms 20824 KB Output is correct
29 Correct 216 ms 20648 KB Output is correct
30 Correct 215 ms 20900 KB Output is correct
31 Correct 214 ms 20888 KB Output is correct
32 Correct 221 ms 20472 KB Output is correct
33 Correct 271 ms 20568 KB Output is correct
34 Correct 304 ms 20824 KB Output is correct
35 Correct 277 ms 20832 KB Output is correct
36 Correct 298 ms 20824 KB Output is correct
37 Correct 472 ms 21080 KB Output is correct
38 Correct 497 ms 22360 KB Output is correct
39 Correct 439 ms 20056 KB Output is correct
40 Correct 453 ms 19548 KB Output is correct
41 Correct 415 ms 19516 KB Output is correct
42 Correct 407 ms 19332 KB Output is correct
43 Correct 398 ms 19292 KB Output is correct
44 Correct 386 ms 19288 KB Output is correct
45 Correct 311 ms 19292 KB Output is correct
46 Correct 175 ms 19292 KB Output is correct
47 Correct 181 ms 19292 KB Output is correct
48 Correct 148 ms 19572 KB Output is correct
49 Execution timed out 2035 ms 172484 KB Time limit exceeded
50 Halted 0 ms 0 KB -