# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
404203 | wmrmr | 친구 (IOI14_friend) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "friend.h"
#include <bits/stdc++.h>
using namespace std;
const int MAX = 1e5;
bool adj[10][10];
int n;
int confidence[MAX], host[MAX], protocol[MAX], dp[MAX][2];
vector<int> g[MAX];
void pt4_DFS(int v, int pai)
{
dp[v][1] += confidence[v];
for( int=0;i<g[v].size();i++ )
{
int prox = g[v][i];
if(prox == pai) continue;
pt4_DFS(prox,v);
dp[v][0] += max( dp[prox][0] , dp[prox][1]);
dp[v][1] += dp[prox][0];
}
}
// Find out best sample
int findSample(int tn,int tconfidence[],int thost[],int tprotocol[]){
int ans = 0;
n = tn;
for(int i=0;i<n;i++) confidence[i] = tconfidence[i], host[i] = thost[i], protocol[i] = tprotocol[i];
bool pt2 = 1, pt3 = 1, pt4 = 1;
for(int i=1;i<n;i++)
{
if(protocol[i] != 0) pt4 = 0;
if(protocol[i] != 1) pt2 = 0;
if(protocol[i] != 2) pt3 = 0;
}
if(n <= 10)
{
ans = 0;
for(int i=1;i<n;i++)
{
int p = host[i];
if(protocol[i] != 1) adj[i][p] = adj[p][i] = 1;
if(protocol[i] != 0)
for(int j=0;j<i;j++)
adj[i][j] = adj[j][i] = adj[j][i] || adj[p][j];
}
int dbg = 0;
for(int cur = 0 ; cur < (1<<n) ; cur++)
{
bool valid = 1;
for(int i=0;i<n;i++) for(int j=i+1;j<n;j++)
{
if( (cur&(1<<i)) && (cur&(1<<j)) )
if(adj[i][j]) valid = 0;
}
int temp = 0;
for(int i=0;i<n;i++)
if(cur&(1<<i)) temp += confidence[i];
if(valid) ans = max(ans,temp), dbg = cur;
}
return ans;
}
if(pt2)
{
for(int i=0;i<n;i++) ans += confidence[i];
return ans;
}
if(pt3)
{
for(int i=0;i<n;i++) ans = max(ans,confidence[i]);
return ans;
}
if(n <= 1e3)
{
for(int i=1;i<n;i++)
{
int p = host[i];
if(protocol[i] != 1) g[i].push_back(p), g[p].push_back(i);
if(protocol[i] != 0)
{
for(int j=0;j<v[p].size();j++)
{
int prox = v[p][j];
g[prox].push_back(i); g[i].push_back(prox);
}
}
}
}
if(pt4)
{
pt4_DFS(0,0);
return max( dp[0][0] , dp[0][1] );
}
return 0;
}