제출 #582378

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
582378		jasmin	Friend (IOI14_friend)	C++14	27 / 100	21 ms	1356 KiB

friend

#include<bits/stdc++.h>
#include "friend.h"
using namespace std;

int subtask1(int n, int c[], int h[], int p[]){
    vector<set<int> > adi(n);
    for(int i=1; i<n; i++){
        if(p[i]==0){
            adi[h[i]].insert(i);
            adi[i].insert(h[i]);
        }
        else if(p[i]==1){
            for(auto u: adi[h[i]]){
                adi[u].insert(i);
                adi[i].insert(u);
            }
        }
        else{
            for(auto u: adi[h[i]]){
                adi[u].insert(i);
                adi[i].insert(u);
            }
            adi[h[i]].insert(i);
            adi[i].insert(h[i]);
        }
    }

    int ans=0;
    for(int i=0; i<(1<<n); i++){
        vector<int> active;
        int mom=0;
        for(int j=0; j<n; j++){
            if((i>>j)%2==1){
                mom+=c[j];
                active.push_back(j);
            }
        }

        for(auto v: active){
            for(auto u: active){
                if(adi[v].find(u)!=adi[v].end()) mom=0;
            }
        }
        ans=max(ans, mom);
    }
    return ans;
}

bool issubtask2(int n, int protocol[]){
    for(int i=1; i<n; i++){
        if(protocol[i]!=1) return false;
    }
    return true;
}
int subtask2(int n, int confidence[]){
    int ans=0;
    for(int i=0; i<n; i++){
        ans+=confidence[i];
    }
    return ans;
}

bool issubtask3(int n, int protocol[]){
    for(int i=1; i<n; i++){
        if(protocol[i]!=2) return false;
    }
    return true;
}
int subtask3(int n, int confidence[]){
    int ans=0;
    for(int i=0; i<n; i++){
        ans=max(ans, confidence[i]);
    }
    return ans;
}

int findSample(int n,int confidence[],int host[],int protocol[]){
    if(n<=10){
        return subtask1(n, confidence, host, protocol);
    }
    if(issubtask2(n, protocol)){
        return subtask2(n, confidence);
    }
    if(issubtask3(n, protocol)){
        return subtask3(n, confidence);
    }
    return -1;
}

/*signed main(){
    ios_base::sync_with_stdio(false);
    cin.tie(0);


}*/

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