Submission #313558

#TimeUsernameProblemLanguageResultExecution timeMemory
313558juggernautFriend (IOI14_friend)C++14
69 / 100
40 ms4352 KiB
#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],timer,used[1005],match[1005];
bool vis[1005];
vector<int>v[2];
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 vv,int color){
    vis[vv]=1;
    v[color].push_back(vv);
	for(int to:g[vv])if(!vis[to])go(to,1-color);
}
bool check(int v){
	used[v]=timer;
	for(int to:g[v]){
		if(match[to]==-1||used[match[to]]!=timer&&check(match[to])) {
			match[to]=v;
			return 1;
		}
	}
	return 0;
}
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{
            memset(match,-1,sizeof(match));
            for(int i=0;i<n;i++)if(!vis[i])go(i,0);
            int cur=0;
            for(int to:v[0]){
                timer++;
                cur+=check(to);
            }
            return n-cur;
        }
    }
}

Compilation message (stderr)

friend.cpp: In function 'bool check(int)':
friend.cpp:32:43: warning: suggest parentheses around '&&' within '||' [-Wparentheses]
   32 |   if(match[to]==-1||used[match[to]]!=timer&&check(match[to])) {
      |                     ~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...