Submission #25408

#TimeUsernameProblemLanguageResultExecution timeMemory
25408kdh9949Election Campaign (JOI15_election_campaign)C++14
100 / 100
606 ms35316 KiB
#include <bits/stdc++.h>
using namespace std;

struct Q{ int x, y, c; };

int n, m, d[100010], dp[100010], ps[100010], pe[100010], cnt, sp[17][100010];
vector<int> e[100010], ch[100010];
vector<Q> q[100010];

const int sz = 131072;
struct Seg{
	int dat[2 * sz];
	void upd(int s, int e, int x){
		for(s += sz, e += sz; s <= e; s /= 2, e /= 2){
			if(s % 2 == 1) dat[s++] += x;
			if(e % 2 == 0) dat[e--] += x;
		}
	}
	int get(int x){
		int ret = 0;
		for(x += sz; x; x /= 2) ret += dat[x];
		return ret;
	}
} S, T;

void g(int x, int p){
	ps[x] = ++cnt;
	sp[0][x] = p;
	d[x] = d[p] + 1;
	for(int i = 1; i < 17; i++) sp[i][x] = sp[i - 1][sp[i - 1][x]];
	for(auto &i : e[x]){
		if(i != p){
			ch[x].push_back(i);
			g(i, x);
		}
	}
	pe[x] = cnt;
}

int lca(int x, int y){
	if(d[x] < d[y]) swap(x, y);
	for(int i = 16; i >= 0; i--) if(d[sp[i][x]] >= d[y]) x = sp[i][x];
	if(x == y) return x;
	for(int i = 16; i >= 0; i--){
		if(sp[i][x] != sp[i][y]){ x = sp[i][x]; y = sp[i][y]; }
	}
	return sp[0][x];
}

void f(int x){
	int cs = 0;
	for(auto &i : ch[x]){
		f(i);
		cs += dp[i];
	}
	for(auto &i : ch[x]) T.upd(ps[i], pe[i], i);
	int ret = cs;
	for(auto &i : q[x]){
		int cret = cs;
		if(i.x != x) cret += S.get(ps[i.x]) - dp[T.get(ps[i.x])];
		if(i.y != x) cret += S.get(ps[i.y]) - dp[T.get(ps[i.y])];
		ret = max(ret, cret + i.c);
	}
	for(auto &i : ch[x]){
		S.upd(ps[x] + 1, ps[i] - 1, dp[i]);
		S.upd(pe[i] + 1, pe[x], dp[i]);
		T.upd(ps[i], pe[i], -i);
	}
	S.upd(ps[x], ps[x], cs);
	dp[x] = ret;
}

int main(){
	scanf("%d", &n);
	for(int i = 0, x, y; i < n - 1; i++){
		scanf("%d%d", &x, &y);
		e[x].push_back(y);
		e[y].push_back(x);
	}
	g(1, 1);
	scanf("%d", &m);
	for(int x, y, z; m--; ){
		scanf("%d%d%d", &x, &y, &z);
		q[lca(x, y)].push_back({x, y, z});
	}
	f(1);
	printf("%d\n", dp[1]);
}

Compilation message (stderr)

election_campaign.cpp: In function 'int main()':
election_campaign.cpp:74:17: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d", &n);
                 ^
election_campaign.cpp:76:24: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d%d", &x, &y);
                        ^
election_campaign.cpp:81:17: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d", &m);
                 ^
election_campaign.cpp:83:30: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d%d%d", &x, &y, &z);
                              ^
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...