Submission #163152

#	Time	Username	Problem	Language	Result	Execution time	Memory
163152		mhy908	Friend (IOI14_friend)	C++14	100 / 100	44 ms	4216 KiB

friend

#include <bits/stdc++.h>
#include "friend.h"
using namespace std;
typedef long long LL;
LL chose[100010], nochose[100010];
int findSample(int n, int con[], int host[], int proto[])
{
    for(int i=0; i<n; i++){
        chose[i]=(LL)con[i];
    }
    for(int i=n-1; i>=1; i--){
        LL c1=chose[host[i]], n1=nochose[host[i]], c2=chose[i], n2=nochose[i];
        if(proto[i]==0){
            chose[host[i]]=c1+n2;
            nochose[host[i]]=max(c2+n1, n1+n2);
        }
        if(proto[i]==1){
            chose[host[i]]=max(c1+c2, max(c1+n2, n1+c2));
            nochose[host[i]]=n1+n2;
        }
        if(proto[i]==2){
            chose[host[i]]=max(c1+n2, c2+n1);
            nochose[host[i]]=n1+n2;
        }
    }
    return (int)max(chose[0], nochose[0]);
}

• Subtask 1: 11 / 11
• Subtask 2: 8 / 8
• Subtask 3: 8 / 8
• Subtask 4: 19 / 19
• Subtask 5: 23 / 23
• Subtask 6: 31 / 31