Submission #1064091

#	Time	Username	Problem	Language	Result	Execution time	Memory
1064091		VMaksimoski008	Friend (IOI14_friend)	C++17	27 / 100	20 ms	1368 KiB

friend

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

int mat[10][10];

int findSample(int n, int confidence[], int host[], int protocol[]) {
	int ans=0;

    if(n <= 10) {
        for(int i=1; i<n; i++) {
            if(protocol[i] == 0) {
                mat[host[i]][i] = mat[i][host[i]] = 1;
            } else if(protocol[i] == 1) {
                for(int j=0; j<n; j++)
                    if(mat[host[i]][j]) mat[i][j] = mat[j][i] = 1;
            } else {
                mat[host[i]][i] = mat[i][host[i]] = 1;
                for(int j=0; j<n; j++)
                    if(mat[host[i]][j]) mat[i][j] = mat[j][i] = 1;
            }
        }

        for(int s=0; s<(1<<n); s++) {
            bool ok = 1;
            vector<int> vec;
            for(int i=0; i<n; i++) {
                if(s & (1 << i)) {
                    for(int &x : vec) if(mat[x][i]) ok = 0;
                    vec.push_back(i);
                }
            }

            if(ok) {
                int sum = 0;
                for(int &x : vec) sum += confidence[x];
                ans = max(ans, sum);
            }
        }
        return ans;
    }

    set<int> st;
    for(int i=1; i<n; i++) st.insert(protocol[i]);

    if(st.size() == 1 && *st.begin() == 1) {
        for(int i=0; i<n; i++) ans += confidence[i];
        return ans;
    }

    if(st.size() == 1 && *st.begin() == 2) {
        for(int i=0; i<n; i++) ans = max(ans, confidence[i]);
        return ans;
    }

	return ans;
}

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