제출 #1178154

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1178154		alexander707070	Tree (IOI24_tree)	C++20	10 / 100	2094 ms	20896 KiB

tree

#include<bits/stdc++.h>
#define MAXN 200007
using namespace std;

int n;
vector<int> v[MAXN];
long long cost[MAXN],ans;

long long L,R,sum[MAXN],dp[MAXN];

void init(vector<int> P, vector<int> W){
	n=int(P.size());

	for(int i=1;i<=n;i++)cost[i]=W[i-1];

	for(int i=1;i<P.size();i++){
		v[P[i]+1].push_back(i+1);
	}
}

void solve(int x){
	sum[x]=0;

	for(int i:v[x]){
		solve(i);
		sum[x]+=sum[i];
	}

	if(sum[x]>R){
		ans+=cost[x]*(sum[x]-R);
		sum[x]=R;
	}

	if(sum[x]<L){
		ans+=cost[x]*(L-sum[x]);
		sum[x]=L;
	}
}

long long query(int l, int r){
	L=l; R=r;
	ans=0;
	solve(1);
	return ans;
}

/*int main(){

	init({-1, 0, 0}, {1, 1, 1});
	cout<<query(1,2)<<"\n";

	return 0;
}*/

• Subtask 0: 0 / 0
• Subtask 1: 10 / 10
• Subtask 2: 0 / 13
• Subtask 3: 0 / 18
• Subtask 4: 0 / 7
• Subtask 5: 0 / 11
• Subtask 6: 0 / 22
• Subtask 7: 0 / 19