oj.uz

Submission #836711

# Submission time Handle Problem Language Result Execution time Memory
836711 2023-08-24T14:24:40 Z Juan Magic Tree (CEOI19_magictree) C++17
0 / 100
 113 ms 16048 KB 
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define tii tuple<int,int,int,int>
#define pii pair<int,int>
#define ff first
#define ss second
#define all(x) x.begin(),x.end()
const int maxn = 1e5 + 5;

vector<int> g[maxn];
int tin[maxn], tout[maxn], pai[maxn], ans_arr[maxn], depth[maxn];
vector<tii> fruits;

int timer=1;
void elr_tr(int u, int p){
	tin[u] = timer++;
	depth[u] = depth[p]+1;

	for(int v : g[u]) if(v!=p){
		elr_tr(v,u);
	}
	tout[u] = timer;
}

int bit[maxn];

void upd(int x, int val){
	for(; x < maxn; x+=x&-x) bit[x] += val;
}
int query(int l, int r){
	int rt = 0;
	for(int i = r; i>0; i-=i&-i) rt += bit[i];
	for(int i = l-1; i>0; i-=i&-i) rt -= bit[i];
	return rt;
}

int solve(int u, int p){
	int aux = 0;
	for(int v : g[u]) if(v!=p){
		aux += solve(v,u);
	}
	return max(ans_arr[u], aux);
}

int32_t main(){
	int n, m, k; cin >> n >> m >> k;
	for(int i = 2; i <= n; i++){
		int p; cin >> p;
		pai[i] = p;
		g[i].push_back(p);
		g[p].push_back(i);
	}

	elr_tr(1,1);

	for(int i = 0; i < m; i++){
		int v, d, w; cin >> v >> d >> w;
		fruits.push_back({d,-depth[v],v,w});
	}

	sort(all(fruits));
	for(int i = 0; i < fruits.size(); i++){
		auto[d,dep,u,w] = fruits[i];
		upd(tin[u], w);
		ans_arr[u] = query(tin[u], tout[u]);
	}

	cout << solve(1,1) << '\n';
}

Compilation message

magictree.cpp: In function 'int32_t main()':
magictree.cpp:63:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::tuple<long long int, long long int, long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   63 |  for(int i = 0; i < fruits.size(); i++){
      |                 ~~^~~~~~~~~~~~~~~

Subtask #1
0 / 6.0

# Verdict Execution time Memory Grader output
1 Incorrect 2 ms 2644 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 3.0

# Verdict Execution time Memory Grader output
1 Incorrect 66 ms 13116 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 2800 KB Output is correct
2 Incorrect 2 ms 2772 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #4
0 / 12.0

# Verdict Execution time Memory Grader output
1 Incorrect 113 ms 16048 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #5
0 / 16.0

# Verdict Execution time Memory Grader output
1 Incorrect 2 ms 2644 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #6
0 / 13.0

# Verdict Execution time Memory Grader output
1 Incorrect 8 ms 4464 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #7
0 / 22.0

# Verdict Execution time Memory Grader output
1 Incorrect 2 ms 2644 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #8
0 / 17.0

# Verdict Execution time Memory Grader output
1 Incorrect 2 ms 2644 KB Output isn't correct
2 Halted 0 ms 0 KB -