Submission #313522

#	Time	Username	Problem	Language	Result	Execution time	Memory
313522		juggernaut	Friend (IOI14_friend)	C++14	46 / 100	38 ms	4352 KiB

friend

#include"friend.h"
//#include"grader.cpp"
#include<bits/stdc++.h>
using namespace std;
vector<int>g[1005];
bool can[15][15];
int dp[1005][2],conf[1005],c1,c2;
bool vis[1005];
void add(int x,int y){
    g[x].push_back(y);
    g[y].push_back(x);
}
void dfs(int v,int p){
    dp[v][0]=dp[v][1]=0;
    dp[v][1]+=conf[v];
    for(int to:g[v])
        if(to!=p){
            dfs(to,v);
            dp[v][0]+=max(dp[to][0],dp[to][1]);
            dp[v][1]+=dp[to][0];
        }
}
void go(int v,int color){
    c1+=color;
    c2+=1-color;
    vis[v]=1;
    for(int to:g[v])if(!vis[to])go(to,1-color);
}
int findSample(int n,int C[],int host[],int protocol[]){
    copy(C,C+n,conf);
    for(int i=1;i<n;i++){
        if(protocol[i]==0){
            add(host[i],i);
        }else if(protocol[i]==1){
            for(int to:g[host[i]])add(i,to);
        }else if(protocol[i]==2){
            for(int to:g[host[i]])add(i,to);
            add(host[i],i);
        }
    }
    if(n<11){
        for(int i=0;i<n;i++)for(int to:g[i])can[i][to]=1,can[to][i]=1;
        int mx=0,sum;
        for(int mask=0;mask<(1<<n);mask++){
            sum=0;
            for(int i=0;i<n;i++)
                if(mask>>i&1){
                    sum+=conf[i];
                    for(int j=0;j<n;j++)if((mask>>j&1)&&can[i][j])sum=-2e9;
                }
            mx=max(mx,sum);
        }
        return mx;
    }else{
        int i,cnt[3]={0,0,0};
        for(i=1;i<n;i++)
            cnt[protocol[i]]++;
        if(cnt[0]==n-1){
            dfs(0,0);
            return max(dp[0][0],dp[0][1]);
        }else if(cnt[1]==n-1){
            int sum=0;
            for(int i=0;i<n;i++)sum+=conf[i];
            return sum;
        }else if(cnt[2]==n-1){
            int sum=0;
            for(int i=0;i<n;i++)sum=max(sum,conf[i]);
            return sum;
        }else{
            int res=0;
            for(int i=0;i<n;i++)
                if(!vis[i]){
                    c1=c2=0;
                    go(i,0);
                    res+=min(c1,c2);
                }
            return n-res;
        }
    }
}

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