Submission #31079

#	Time	Username	Problem	Language	Result	Execution time	Memory
31079		WhipppedCream	Friend (IOI14_friend)	C++14	100 / 100	43 ms	5144 KiB

friend

#include <bits/stdc++.h>
#define ii pair<int, int>
#define X first
#define Y second
#define pb push_back
#define mp make_pair
#define vi vector<int>
#define vii vector< pair<int, int> >
typedef long long ll;
using namespace std;
int Host[100005];
int Prot[100005];
int conf[100005];
int c[100005];
int nc[100005];
int findSample(int n, int *conf, int *Host, int *Prot)
{
  for(int i = 0; i< n; i++) c[i] = conf[i];
    for(int i = n-1; i>= 1; i--)
    {
        int x = Host[i];
        if(Prot[i] == 1)
        {
            c[x] = max(max(c[x]+nc[i], c[i]+nc[x]), c[i]+c[x]);
            nc[x] = nc[x] + nc[i];
        }
        else if(Prot[i] == 2)
        {
            c[x] = max(c[x]+nc[i], c[i]+nc[x]);
            nc[x] = nc[x]+nc[i];
        }
        else
        {
            c[x] = c[x]+nc[i];
            nc[x] = max(c[i]+nc[x], nc[i]+nc[x]);
        }
    }
  return max(c[0], nc[0]);
	return 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