# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
7581 | jihoon | 친구 (IOI14_friend) | C++98 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<cstdio>
#include<algorithm>
using namespace std;
int n;
int confi[100001];
int host[100001];
int protocol[100001];
int dp[100001][2];
int main(){
int i;
scanf("%d",&n);
for(i=0;i<n;i++){
scanf("%d",&confi[i]);
dp[i][0]=confi[i];
}
for(i=1;i<n;i++){
scanf("%d",&host[i]);
scanf("%d",&protocol[i]);
}
for(i=n-1;i>=1;i--){
if(protocol[i]==0){
dp[host[i]][0]+=dp[i][1];
dp[host[i]][1]=max(dp[host[i]][1]+dp[i][1],dp[host[i]][1]+dp[i][0]);
}else if(protocol[i]==1){
dp[host[i]][0]=max(dp[host[i]][0]+dp[i][0],max(dp[host[i]][0]+dp[i][1],dp[host[i]][1]+dp[i][0]));
dp[host[i]][1]+=dp[i][1];
}else{
dp[host[i]][0]=max(dp[host[i]][0]+dp[i][1],dp[host[i]][1]+dp[i][0]);
dp[host[i]][1]+=dp[i][1];
}
}
printf("%d",max(dp[0][0],dp[0][1]));
}