Submission #589923

#TimeUsernameProblemLanguageResultExecution timeMemory
589923Dan4LifeFriend (IOI14_friend)C++17
0 / 100
1090 ms5004 KiB
#include <bits/stdc++.h>
//#define LOL
#if !defined(LOL)
    #include "friend.h"
#endif
using namespace std;
 
#define pb push_back
#define sz(a) (int)a.size()
#define all(a) a.begin(),a.end()
const int maxn = (int)2e5+10;
vector<int> adj[maxn]; 
set<int> S;
void add_edge(int a, int b){
    adj[a].pb(b); adj[b].pb(a);
}

bool chk(int s){
    for(auto u : adj[s])
        if(S.count(u)) return false;
    return true;
}

int findSample(int n, int c[], int h[], int pro[])
{
    for(int i = 1; i < n; i++){
        if(pro[i]==1 or pro[i]==3)
            add_edge(h[i],i);
        else if(pro[i]!=1)
            for(auto u : adj[h[i]]) add_edge(u,i);
    }
    vector<int> v(n-1,0); int cnt = 0;
    for(auto &u : v) u = cnt++; 
    //iota(all(v),0); 
    int ans = 0;
    do{
        int sum = 0; S.clear();
        for(auto u : v){
            if(chk(u)){
                sum+=c[u], S.insert(u);
                for(auto w : adj[u]) S.insert(w);
            }
        }
        ans = max(ans, sum);
    }while(next_permutation(all(v)));
    return ans;
}

#if defined(LOL)

int main(void)
{
    #define __MAXSIZE__ 100002
    int confidence[__MAXSIZE__];
    int host[__MAXSIZE__];
    int protocol[__MAXSIZE__];
    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;
}
#endif
#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...