Submission #153363

#	Time	Username	Problem	Language	Result	Execution time	Memory
153363		arnold518	Friend (IOI14_friend)	C++14	100 / 100	41 ms	3448 KiB

friend

#include "friend.h"
#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;

const int MAXN = 1e5;

int N, *C, *H, *P;
pii dp[MAXN+10];

int findSample(int _N, int *_C, int *_H, int *_P)
{
    int i, j;
    N=_N; C=_C; H=_H; P=_P;

    for(i=0; i<N; i++) dp[i]={C[i], 0};

    for(i=N-1; i>=1; i--)
    {
        if(P[i]==0) dp[H[i]]={dp[H[i]].first+dp[i].second, max(dp[i].first+dp[H[i]].second, dp[i].second+dp[H[i]].second)};
        else if(P[i]==1) dp[H[i]]={max({dp[H[i]].first+dp[i].second, dp[i].first+dp[H[i]].second, dp[i].first+dp[H[i]].first}), dp[i].second+dp[H[i]].second};
        else dp[H[i]]={max(dp[H[i]].first+dp[i].second, dp[i].first+dp[H[i]].second), dp[i].second+dp[H[i]].second};
    }
    return max(dp[0].first, dp[0].second);
}

Compilation message (stderr)

friend.cpp: In function 'int findSample(int, int*, int*, int*)':
friend.cpp:16:12: warning: unused variable 'j' [-Wunused-variable]
     int i, j;
            ^

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