oj.uz

Submission #253367

# Submission time Handle Problem Language Result Execution time Memory
253367 2020-07-27T19:03:55 Z eohomegrownapps Friend (IOI14_friend) C++14
46 / 100
 35 ms 1656 KB 
#include "friend.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
// Find out best sample
vector<vector<int>> adjlist;
int subtask1(int n,int confidence[],int host[],int protocol[]){
    adjlist.resize(n);
    for (int i = 1; i<n; i++){
        if (protocol[i]==0){
            adjlist[host[i]].push_back(i);
            adjlist[i].push_back(host[i]);
        } else if (protocol[i]==1){
            for (int x : adjlist[host[i]]){
                adjlist[i].push_back(x);
                adjlist[x].push_back(i);
            }
        } else {
            for (int x : adjlist[host[i]]){
                adjlist[i].push_back(x);
                adjlist[x].push_back(i);
            };
            adjlist[host[i]].push_back(i);
            adjlist[i].push_back(host[i]);
        }
    }
    ll mx = 0;
    for (int i = 0; i<(1<<n); i++){
        //check if ok
        bool works = true;
        ll tot = 0;
        for (int j = 0; j<n; j++){
            if ((1<<j)&i){
                tot+=confidence[j];
                for (int k : adjlist[j]){
                    if ((1<<k)&i){
                        works=false;
                        break;
                    }
                }
            }
            if (works==false){
                break;
            }
        }
        if (works==false){
            continue;
        }
        mx=max(mx,tot);
    }
    return mx;
}

int subtask2(int n,int confidence[],int host[],int protocol[]){
    int tot = 0;
    for (int i = 0; i<n; i++){
        tot+=confidence[i];
    }
    return tot;
}

int subtask3(int n,int confidence[],int host[],int protocol[]){
    int mx = 0;
    for (int i = 0; i<n; i++){
        mx=max(mx,confidence[i]);
    }
    return mx;
}

vector<vector<int>> dp;
int *conf;

int f(int node, int par, bool use){
    if (dp[node][use]!=-1){
        return dp[node][use];
    }
    if (use){
        int tot = conf[node];
        for (int i : adjlist[node]){
            if (i==par){continue;}
            tot+=f(i,node,false);
        }
        return dp[node][use]=tot;
    } else {
        int tot = 0;
        for (int i : adjlist[node]){
            if (i==par){continue;}
            tot+=max(f(i,node,true),f(i,node,false));
        }
        return dp[node][use]=tot;
    }
}

int subtask4(int n,int confidence[],int host[],int protocol[]){
    adjlist.resize(n);
    conf = confidence;
    for (int i = 1; i<n; i++){
        adjlist[i].push_back(host[i]);
        adjlist[host[i]].push_back(i);
    }
    dp.resize(n,vector<int>(2,-1));
    return max(f(0,-1,false),f(0,-1,true));
}

int subtask5(int n,int confidence[],int host[],int protocol[]){
    return 0;
}

int subtask6(int n,int confidence[],int host[],int protocol[]){
    return 0;
}

int findSample(int n,int confidence[],int host[],int protocol[]){
	if (n<=10){
        return subtask1(n,confidence,host,protocol); 
    } else {
        bool protocols[3]={0,0,0};
        for (int i = 1; i<n; i++){
            protocols[protocol[i]]=true;
        }
        //cout<<protocols[0]<<' '<<protocols[1]<<' '<<protocols[2]<<'\n';
        if (protocols[0]==0&&protocols[1]==1&&protocols[2]==0){
            return subtask2(n,confidence,host,protocol); 
        } else if (protocols[0]==0&&protocols[1]==0&&protocols[2]==1){
            return subtask3(n,confidence,host,protocol); 
        } else if (protocols[0]==1&&protocols[1]==0&&protocols[2]==0){
            return subtask4(n,confidence,host,protocol); 
        } else if (protocols[0]==1&&protocols[1]==1&&protocols[2]==0){
            return subtask5(n,confidence,host,protocol); 
        } else {
            return subtask6(n,confidence,host,protocol); 
        }
    }
	return 0;
}

Subtask #1
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 384 KB Output is correct
2 Correct 1 ms 384 KB Output is correct
3 Correct 1 ms 384 KB Output is correct
4 Correct 0 ms 384 KB Output is correct
5 Correct 1 ms 384 KB Output is correct
6 Correct 0 ms 384 KB Output is correct
7 Correct 0 ms 384 KB Output is correct
8 Correct 0 ms 384 KB Output is correct
9 Correct 0 ms 384 KB Output is correct
10 Correct 0 ms 384 KB Output is correct
11 Correct 0 ms 384 KB Output is correct
12 Correct 0 ms 384 KB Output is correct
13 Correct 0 ms 384 KB Output is correct
14 Correct 0 ms 384 KB Output is correct
15 Correct 0 ms 384 KB Output is correct
16 Correct 0 ms 384 KB Output is correct
17 Correct 0 ms 384 KB Output is correct

Subtask #2
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 384 KB Output is correct
2 Correct 0 ms 384 KB Output is correct
3 Correct 1 ms 384 KB Output is correct
4 Correct 1 ms 384 KB Output is correct
5 Correct 1 ms 384 KB Output is correct
6 Correct 1 ms 384 KB Output is correct
7 Correct 1 ms 384 KB Output is correct
8 Correct 1 ms 384 KB Output is correct
9 Correct 1 ms 384 KB Output is correct
10 Correct 1 ms 384 KB Output is correct

Subtask #3
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 384 KB Output is correct
2 Correct 1 ms 384 KB Output is correct
3 Correct 0 ms 384 KB Output is correct
4 Correct 1 ms 384 KB Output is correct
5 Correct 1 ms 384 KB Output is correct
6 Correct 0 ms 384 KB Output is correct
7 Correct 0 ms 384 KB Output is correct
8 Correct 0 ms 384 KB Output is correct
9 Correct 0 ms 384 KB Output is correct
10 Correct 1 ms 384 KB Output is correct

Subtask #4
19.0 / 19.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 384 KB Output is correct
2 Correct 0 ms 384 KB Output is correct
3 Correct 0 ms 384 KB Output is correct
4 Correct 0 ms 384 KB Output is correct
5 Correct 1 ms 512 KB Output is correct
6 Correct 1 ms 512 KB Output is correct
7 Correct 1 ms 384 KB Output is correct
8 Correct 1 ms 512 KB Output is correct
9 Correct 0 ms 384 KB Output is correct
10 Correct 1 ms 384 KB Output is correct
11 Correct 1 ms 384 KB Output is correct
12 Correct 1 ms 512 KB Output is correct
13 Correct 1 ms 512 KB Output is correct
14 Correct 1 ms 384 KB Output is correct
15 Correct 1 ms 512 KB Output is correct

Subtask #5
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 384 KB Output is correct
2 Correct 0 ms 384 KB Output is correct
3 Correct 1 ms 384 KB Output is correct
4 Correct 0 ms 384 KB Output is correct
5 Correct 0 ms 384 KB Output is correct
6 Correct 0 ms 384 KB Output is correct
7 Correct 0 ms 384 KB Output is correct
8 Correct 1 ms 384 KB Output is correct
9 Correct 1 ms 512 KB Output is correct
10 Correct 1 ms 384 KB Output is correct
11 Incorrect 1 ms 384 KB Output isn't correct
12 Halted 0 ms 0 KB -

Subtask #6
0 / 31.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 384 KB Output is correct
2 Correct 0 ms 384 KB Output is correct
3 Correct 1 ms 384 KB Output is correct
4 Correct 0 ms 384 KB Output is correct
5 Correct 1 ms 384 KB Output is correct
6 Correct 0 ms 384 KB Output is correct
7 Correct 0 ms 384 KB Output is correct
8 Correct 0 ms 384 KB Output is correct
9 Correct 0 ms 384 KB Output is correct
10 Correct 1 ms 256 KB Output is correct
11 Correct 1 ms 384 KB Output is correct
12 Incorrect 35 ms 1656 KB Output isn't correct
13 Halted 0 ms 0 KB -