| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 586356 | LastRonin | 친구 (IOI14_friend) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "friend.h"
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const ll N = 1e3 + 10;
vector<ll> g[N];
int findSample(int n,int confidence[],int host[],int protocol[]){
bool eq = 1;
for(int j = 1; j < n - 1; j++) {
if(protocol[j] != protocol[j + 1])
eq = 0;
}
if(eq) {
if(protocol[1] == 1) {
// empty
ll sum = 0;
for(int j = 0; j < n; j++)
sum += confidence[j];
return sum;
} else if(protocol[1] == 2) {
// full
ll mx = 0;
for(int j = 0; j < n; j++)
mx = max(mx, (ll)confidence[j]);
return mx;
} else if(protocol[1] == 0) {
// tree
return -1;
}
} else {
for(int j = 1; j < n; j++) {
if(protocol[j] == 0) {
g[j].pb(host[j]);
g[host[j]].pb(j);
} else if(protocol[j] == 1) {
for(auto u : g[host[j]]) {
g[j].pb(u);
g[u].pb(j);
}
} else if(protocol[j] == 2) {
for(auto u : g[host[j]]) {
g[j].pb(u);
g[u].pb(j);
}
g[j].pb(host[j]);
g[host[j]].pb(j);
}
}
assert(n <= 10);
ll mx = 0;
for(int j = 0; j < (1<<n); j++) {
bool incor = 0;
ll sum = 0;
for(int i = 0; i < n; i++) {
if((1<<i)&j)
sum = sum + confidence[i];
}
for(int i = 0; i < n; i++) {
if((1<<i)&j) {
for(auto u : g[i])
if((1<<u)&j)
incor = 1;
}
}
if(incor == 0)
mx = max(mx, sum);
}
return mx;
}
return 0;
}
/*
6
13 3 6 20 10 15
0 0 0 1 1 2 2 1 0 0
6
13 3 6 20 10 15
0 0 0 1 1 2 2 1 0 0
*/
/*
#define __MAXSIZE__ 100002
#include "friend.h"
using namespace std;
// Confidence
int confidence[__MAXSIZE__];
// Host
int host[__MAXSIZE__];
// Protocol
int protocol[__MAXSIZE__];
// Main
int main(void)
{
int n,i;
// Number of people
assert(scanf("%d",&n)==1);
// Confidence
for(i=0;i<n;i++)
assert(scanf("%d",&confidence[i])==1);
// Host and Protocol
for(i=1;i<n;i++)
assert(scanf("%d %d",&host[i],&protocol[i])==2);
// Answer
printf("%d\n",findSample(n,confidence,host,protocol));
return 0;
}
/**/