oj.uz

Submission #342550

# Submission time Handle Problem Language Result Execution time Memory
342550 2021-01-02T11:01:59 Z Nursik Election Campaign (JOI15_election_campaign) C++17
10 / 100
 277 ms 86128 KB 
#include <bits/stdc++.h>
          
#define fi first
#define se second
#define pp pop_back
#define ll long long
#define pb push_back
#define ld long double
#define debug cout << "OK\n";
#define all(x) x.begin(), x.end() 
#define mp make_pair 
using namespace std;        
 
const ll N = 1e6 + 200;
const ll mod = 1e9 + 7;
const ll mod2 = 998244353;
const ll pe = mod2; 
const ll pe2 = 570210983;
const ld eps = 1e-6; 
 
 
/*
Rucode: jaqVYNrpMj
JUDGE_ID: 295965SY
dl:160532
*/
void data() {
	#ifdef NURS
        freopen("main.in", "r", stdin);
        freopen("main.out", "w", stdout);
    #endif	
} 

#define int ll
int n, m, timer;
int tin[N], tout[N], up[20][N];
ll dp[N], dp2[N], l[N], r[N], w[N];
vector<int> g[N], g2[N];
struct fenw
{
	ll f[N * 4];
	void upd(int pos, int val)
	{
		for (int i = pos; i <= N; i = (i | (i + 1)))
			f[i] += val;
	}
	ll sum(int pos)
	{
		ll res = 0;
		for (int i = pos; i >= 1; i = (i & (i + 1)) - 1)
			res += f[i];
		return res;
	}
	ll get(int l, int r)
	{
		return sum(r) - sum(l - 1);
	}
};
fenw tr, tr2;
void dfs(int v, int p)
{
	tin[v] = ++timer;
	up[0][v] = p;
	for (int i = 1; i <= 20; i++)
		up[i][v] = up[i - 1][up[i - 1][v]];

	for (int i = 0; i < g[v].size(); i++)
	{
		int to = g[v][i];
		if (to == p)
			continue;
		dfs(to, v);
	}
	tout[v] = ++timer;
}
bool upper(int a, int b)
{
	return (tin[a] <= tin[b] && tout[b] <= tout[a]);
}
int lca(int a, int b)
{
	if (upper(a, b))
		return a;
	if (upper(b, a))
		return b;
	for (int i = 20; i >= 0; i--)
	{
		if (up[i][a] && !upper(up[i][a], b))
			a = up[i][a];
	}
	return up[0][a];
}    
void dfs2(int v, int p)
{
	for (int i = 0; i < g[v].size(); i++)
	{
		int to = g[v][i];
		if (to == p)
			continue;
		dfs2(to, v);
	   	dp[v] += dp2[to];
	}
	dp2[v] = dp[v];
	for (int i = 0; i < g2[v].size(); i++)
	{
		int pos = g2[v][i];
		int a = l[pos], b = r[pos];
		dp2[v] = max(dp2[v] , dp[v] + tr.get(tin[v], tin[a]) + tr.get(tin[v], tin[b]) 
					 				- tr2.get(tin[v], tin[a]) - tr2.get(tin[v], tin[b]) + w[pos]); 
	}
	tr.upd(tin[v], dp[v]);
	tr.upd(tout[v], -dp[v]);
	tr2.upd(tin[v], dp2[v]);
	tr2.upd(tout[v], -dp2[v]);
}
main()
{
	data();
	ios_base::sync_with_stdio(0),
	cin.tie(0),cout.tie(0);
	cin >> n;
	for (int i = 1; i < n; i++)
	{
		int u, v;
		cin >> u >> v;
		g[u].pb(v), g[v].pb(u);
	}
	dfs(1, 0);
	cin >> m;
	for (int i = 1; i <= m; i++)
	{
		cin >> l[i] >> r[i] >> w[i];
		int lc = lca(l[i], r[i]);
		g2[lc].pb(i);
	}
	dfs2(1, 0);
	cout << dp2[1];
}

Compilation message

election_campaign.cpp: In function 'void dfs(long long int, long long int)':
election_campaign.cpp:67:20: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   67 |  for (int i = 0; i < g[v].size(); i++)
      |                  ~~^~~~~~~~~~~~~
election_campaign.cpp: In function 'void dfs2(long long int, long long int)':
election_campaign.cpp:95:20: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   95 |  for (int i = 0; i < g[v].size(); i++)
      |                  ~~^~~~~~~~~~~~~
election_campaign.cpp:104:20: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  104 |  for (int i = 0; i < g2[v].size(); i++)
      |                  ~~^~~~~~~~~~~~~~
election_campaign.cpp: At global scope:
election_campaign.cpp:116:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
  116 | main()
      |      ^
election_campaign.cpp: In function 'void dfs(long long int, long long int)':
election_campaign.cpp:65:12: warning: iteration 19 invokes undefined behavior [-Waggressive-loop-optimizations]
   65 |   up[i][v] = up[i - 1][up[i - 1][v]];
      |   ~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
election_campaign.cpp:64:20: note: within this loop
   64 |  for (int i = 1; i <= 20; i++)
      |                  ~~^~~~~

Subtask #1
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 34 ms 47596 KB Output is correct
2 Correct 33 ms 47700 KB Output is correct
3 Correct 35 ms 47608 KB Output is correct
4 Incorrect 33 ms 47872 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #2
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 33 ms 47596 KB Output is correct
2 Correct 33 ms 47596 KB Output is correct
3 Correct 36 ms 47980 KB Output is correct
4 Correct 156 ms 85996 KB Output is correct
5 Correct 159 ms 85996 KB Output is correct
6 Correct 157 ms 85996 KB Output is correct
7 Correct 158 ms 85996 KB Output is correct
8 Correct 159 ms 86024 KB Output is correct
9 Correct 157 ms 85996 KB Output is correct
10 Correct 155 ms 85900 KB Output is correct

Subtask #3
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 33 ms 47596 KB Output is correct
2 Correct 33 ms 47596 KB Output is correct
3 Correct 36 ms 47980 KB Output is correct
4 Correct 156 ms 85996 KB Output is correct
5 Correct 159 ms 85996 KB Output is correct
6 Correct 157 ms 85996 KB Output is correct
7 Correct 158 ms 85996 KB Output is correct
8 Correct 159 ms 86024 KB Output is correct
9 Correct 157 ms 85996 KB Output is correct
10 Correct 155 ms 85900 KB Output is correct
11 Correct 42 ms 48876 KB Output is correct
12 Correct 163 ms 86128 KB Output is correct
13 Correct 161 ms 85940 KB Output is correct
14 Correct 164 ms 86000 KB Output is correct
15 Correct 160 ms 85996 KB Output is correct
16 Correct 162 ms 85996 KB Output is correct
17 Correct 162 ms 85972 KB Output is correct
18 Correct 161 ms 85996 KB Output is correct
19 Correct 161 ms 85996 KB Output is correct
20 Correct 162 ms 85924 KB Output is correct

Subtask #4
0 / 30.0

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

Subtask #5
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 34 ms 47596 KB Output is correct
2 Correct 33 ms 47700 KB Output is correct
3 Correct 35 ms 47608 KB Output is correct
4 Incorrect 33 ms 47872 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #6
0 / 40.0

# Verdict Execution time Memory Grader output
1 Correct 34 ms 47596 KB Output is correct
2 Correct 33 ms 47700 KB Output is correct
3 Correct 35 ms 47608 KB Output is correct
4 Incorrect 33 ms 47872 KB Output isn't correct
5 Halted 0 ms 0 KB -