oj.uz

답안 #765976

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
765976 2023-06-25T08:24:04 Z vjudge1 Tree Rotations (POI11_rot) C++17
63 / 100
 215 ms 30512 KB 
#include <bits/stdc++.h>
 
#define ios ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define file(s) if (fopen(s".in", "r")) freopen(s".in", "r", stdin), freopen(s".out", "w", stdout)
#define all(a) a.begin() , a.end()
#define F first
#define S second
 
using namespace std;
using ll = long long;
 
const int N = 5e5+5 , inf = 2e9 + 7;
const ll INF = 1e18 ,   mod = 1e9+7;	

int ans[N] , n , mxson[N] , sz[N] , t[N] , m; 
vector<int> g[N];

void precalc(int v , int pr){
	if(v <= n) sz[v] = 1;
	for(int to : g[v]){
		if(to != pr){
			precalc(to,v);
			sz[v] += sz[to];
			if(sz[to] > sz[mxson[v]]) mxson[v] = to;
		}
	}
}

void upd(int i, int x){
    for(; i <= n; i |= (i + 1)) t[i] += x;
}

int get(int i){
    int res = 0;
    for(; i > 0; i = (i & (i + 1)) - 1) res += t[i];
    return res;
}
int ob;

pair<int,int> calc(int v, int pr){
	if(v <= n) {
		int num = get(v);
		return {num,ob-num};
	}
	int res = 0 , res1 = 0;
	for(int to : g[v]){
		 if(to != pr) {
		 	pair<int,int> a = calc(to,v);
		 	res += a.F;
		 	res1 += a.S;
		 }
	}
	return {res,res1};
}
void del(int v, int pr){
	if(v <= n) upd(v,-1);
	for(int to : g[v]) if(to != pr) del(to,v);
}
void add(int v, int pr){
	if(v <= n) upd(v,1);
	for(int to : g[v]) if(to != pr) add(to,v);
}
void dfs(int v , int pr){
	for(int to : g[v]){
		if(to != pr && to != mxson[v]){
			dfs(to,v);
			del(to,v);
		}
	}
	if(mxson[v]) {
		dfs(mxson[v],v);
		ans[v] += ans[mxson[v]];
	}
	int res = 0 , res1 = 0;
	for(int to : g[v]){
		ob = get(n);
		if(to != pr && to != mxson[v]) {
			pair<int,int> b = calc(to,v);
			res += b.F;
			res1 += b.S;
			add(to,v);
			ans[v] += ans[to];
		}
	}
	if(v <= n) upd(v,1);
	else ans[v] += min(res,res1);
} 
int NUM;
void go(int v){
	int x;
	cin >> x;
	if(x == 0){
		int to = ++NUM;
		g[v].push_back(to);
		g[to].push_back(v);
		go(to);
	} else {
		g[v].push_back(x);
		g[x].push_back(v);
	}
	cin >> x;
	if(x == 0){
		int to = ++NUM;
		g[v].push_back(to);
		g[to].push_back(v);
		go(to);
	} else {
		g[v].push_back(x);
		g[x].push_back(v);
	}
}
void solve(){
	cin >> n;
	NUM = n+1;
	int a; cin >> a;
	go(n+1);
	precalc(n+1,0);
	dfs(n+1,0);
	cout << ans[n+1] << "\n";
}
/*

*/
signed main(){
	ios;
	solve();
	return 0;
}

Subtask #1
9.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 8 ms 12116 KB Output is correct
2 Correct 6 ms 11988 KB Output is correct
3 Correct 5 ms 11988 KB Output is correct
4 Correct 6 ms 11988 KB Output is correct
5 Correct 5 ms 11988 KB Output is correct

Subtask #2
9.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 5 ms 11988 KB Output is correct
2 Correct 6 ms 11988 KB Output is correct
3 Correct 6 ms 12104 KB Output is correct
4 Correct 5 ms 11988 KB Output is correct

Subtask #3
9.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 6 ms 12116 KB Output is correct
2 Correct 6 ms 12156 KB Output is correct
3 Correct 6 ms 12116 KB Output is correct
4 Correct 6 ms 12120 KB Output is correct

Subtask #4
9.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 7 ms 12628 KB Output is correct
2 Correct 8 ms 12436 KB Output is correct
3 Correct 8 ms 12664 KB Output is correct
4 Correct 7 ms 12720 KB Output is correct

Subtask #5
9.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 11 ms 14036 KB Output is correct
2 Correct 18 ms 13140 KB Output is correct
3 Correct 36 ms 14744 KB Output is correct
4 Correct 14 ms 13936 KB Output is correct

Subtask #6
9.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 50 ms 16256 KB Output is correct
2 Correct 34 ms 18132 KB Output is correct
3 Correct 39 ms 19940 KB Output is correct
4 Correct 41 ms 20044 KB Output is correct

Subtask #7
9.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 46 ms 26524 KB Output is correct
2 Correct 51 ms 23680 KB Output is correct
3 Correct 63 ms 21508 KB Output is correct
4 Correct 51 ms 20504 KB Output is correct

Subtask #8
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 76 ms 22752 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #9
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 215 ms 30512 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #10
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 132 ms 29856 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #11
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 193 ms 30192 KB Output isn't correct
2 Halted 0 ms 0 KB -