oj.uz

Submission #1092448

# Submission time Handle Problem Language Result Execution time Memory
1092448 2024-09-24T06:23:59 Z alexander707070 Fireworks (APIO16_fireworks) C++14
0 / 100
 2000 ms 2652 KB 
#include<bits/stdc++.h>
#define MAXN 100007
using namespace std;

const long long inf=1e17;

int n,m;
int p[MAXN],c[MAXN];
vector< pair<int,long long> > v[MAXN];
vector<long long> dists;

long long sum[MAXN],sz[MAXN],ans;

void dfs(int x,long long d){
	sz[x]=(v[x].empty());

	if(sz[x]==1)dists.push_back(d);

	for(int i=0;i<v[x].size();i++){
		dfs(v[x][i].first,d+v[x][i].second);

		sz[x]+=sz[v[x][i].first];
		sum[x]+=sum[v[x][i].first]+v[x][i].second*sz[v[x][i].first];
	}
}

long long solve(int x,long long len,long long e){
	long long res=0;
	long long diff=sum[x]+sz[x]*e-sz[x]*len;
	
	while(diff<-sz[x]/2){diff+=sz[x]; e++; res++;}
	while(diff>sz[x]/2 and e>0){diff-=sz[x]; e--; res++;}

	for(int i=0;i<v[x].size();i++){
		res+=solve(v[x][i].first,len-e,v[x][i].second);
	}

	return res;
}

int main(){
	
	cin>>n>>m;
	n+=m;
	for(int i=2;i<=n;i++){
		cin>>p[i]>>c[i];
		v[p[i]].push_back({i,c[i]});
	}

	dfs(1,0);

	ans=inf;
	for(long long len:dists){
		long long curr=0;
		for(int i=0;i<v[1].size();i++){
			curr+=solve(v[1][i].first,len,v[1][i].second);
		}

		ans=min(ans,curr);
	}

	cout<<ans<<"\n";

    return 0;
}

Compilation message

fireworks.cpp: In function 'void dfs(int, long long int)':
fireworks.cpp:19:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   19 |  for(int i=0;i<v[x].size();i++){
      |              ~^~~~~~~~~~~~
fireworks.cpp: In function 'long long int solve(int, long long int, long long int)':
fireworks.cpp:34:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   34 |  for(int i=0;i<v[x].size();i++){
      |              ~^~~~~~~~~~~~
fireworks.cpp: In function 'int main()':
fireworks.cpp:55:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   55 |   for(int i=0;i<v[1].size();i++){
      |               ~^~~~~~~~~~~~

Subtask #1
0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 2648 KB Output is correct
2 Execution timed out 2045 ms 2652 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #2
0 / 19.0

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

Subtask #3
0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 2648 KB Output is correct
2 Execution timed out 2045 ms 2652 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #4
0 / 45.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 2648 KB Output is correct
2 Execution timed out 2045 ms 2652 KB Time limit exceeded
3 Halted 0 ms 0 KB -