Submission #50781

#	Time	Username	Problem	Language	Result	Execution time	Memory
50781		Just_Solve_The_Problem	Friend (IOI14_friend)	C++11	0 / 100	8 ms	6640 KiB

friend

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

using namespace std;

#define pb push_back
#define sz(s) (int)s.size()
#define pii pair < int, int >
#define fr first
#define sc second
#define mk make_pair
#define all(s) s.begin(), s.end()

const int N = (int)1e5 + 7;
const int smN = (int)1e3 + 7;

vector < int > gr[N];
int has[smN][smN];
int c[N], dp[N], used[N];

void addedge(int a, int b) {
  gr[a].pb(b);
  gr[b].pb(a);
  has[a][b] = has[b][a] = 1;
}

void dfs(int v, int pr) {
  used[v] = 1;
  for (int to : gr[v]) {
    if (to == pr) continue;
    assert(used[to] == 0);
    dfs(to, v);
    if (v != pr) {
      dp[pr] = max(dp[pr], dp[to] + c[pr]);
    }
    dp[v] = max(dp[v], dp[to]);
  }
}


int findSample(int n, int confidence[], int host[], int protocol[]){
  for (int i = 0; i < n; i++) c[i] = confidence[i];
  int ans = 0;
  for (int i = 1; i < n; i++) {
    if (protocol[i] == 0 || protocol[i] == 2) {
      addedge(host[i], i);
    }
    if (protocol[i] == 1 || protocol[i] == 2) {
      for (int to : gr[host[i]]) {
        addedge(to, i);
      }
    }
  }
  int st = -1;
  for (int i = 0; i < n; i++) {
    dp[i] = c[i];
    if (sz(gr[i]) == 1) st = i;
  }
  assert(st != -1);
  dfs(st, st);
  for (int i = 0; i < n; i++) {
    assert(used[i] == 0);
  }
	return dp[st];
}

Compilation message (stderr)

friend.cpp: In function 'int findSample(int, int*, int*, int*)':
friend.cpp:43:7: warning: unused variable 'ans' [-Wunused-variable]
   int ans = 0;
       ^~~

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