제출 #392878

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
392878		Aldas25	친구 (IOI14_friend)	C++14	35 / 100	3 ms	2764 KiB

friend

#include "friend.h"
#include <bits/stdc++.h>

using namespace std;

#define FOR(i, a, b) for (int i = (a); i <= (b); i++)
#define REP(n) FOR(O, 1, (n))
#define f first
#define s second
#define pb push_back
typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<pii> vii;

const int MAXN = 100100;

int n;
int conf[MAXN], h[MAXN], prot[MAXN];

int sub2 () {
    int ret = 0;
    FOR(i, 0, n-1) ret += conf[i];
    return ret;
}

int sub3 () {
    int ret = 0;
    FOR(i, 0, n-1) ret = max(ret, conf[i]);
    return ret;
}

int dp[MAXN][2];
vi adj[MAXN];

void dfs (int v, int p) {
    dp[v][1] = conf[v];
    for (int u : adj[v]) {
        if (u == p) continue;
        dfs (u, v);
        dp[v][0] += max(dp[u][0], dp[u][1]);
        dp[v][1] += dp[u][0];
    }
}

int sub4 () {
    FOR(i, 1, n-1) adj[h[i]].pb(i);
    dfs (0, -1);
    return max(dp[0][1], dp[0][0]);
}

int findSample(int N,int confidence[],int host[],int protocol[]){
    n = N;
    FOR(i, 0, n-1) conf[i] = confidence[i];
    FOR(i, 1, n-1) h[i] = host[i], prot[i] = protocol[i];

    bool is2 = true;
    bool is3 = true;
    bool is4 = true;
    FOR(i, 1, n-1) {
        if (protocol[i] != 1) is2 = false;
        if (protocol[i] != 2) is3 = false;
        if (protocol[i] != 0) is4 = false;
    }

    if (is2)
        return sub2();
    else if (is3)
        return sub3();
    else if (is4)
        return sub4();
    else
        return sub4();
}

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