oj.uz

Submission #835350

# Submission time Handle Problem Language Result Execution time Memory
835350 2023-08-23T13:37:36 Z Liudas Friend (IOI14_friend) C++17
27 / 100
 1000 ms 15060 KB 
#include <cstdio>
#include <cassert>
#define __MAXSIZE__ 100002
#include "friend.h"
#include <vector>
#include <set>
#include <algorithm>
#include <numeric>
#define __MAXSIZE__ 100002
using namespace std;
void getmax(int head, int N, vector<vector<int>> &arr, int &m, set<int> been, int score, int conf[]){
    if(head == N){
        m = max(m, score);
        return;
    }
    set<int> nbeen = been;
    for(int i : arr[head]){
        nbeen.insert(i);
    }
    getmax(head + 1, N, arr, m, been, score, conf);
    if(been.find(head) == been.end())
    getmax(head + 1, N, arr, m, nbeen, score + conf[head], conf);
}
void dfs(int head, int par, vector<vector<int>> &arr, vector<int> &val, vector<int> &bel){
    int v = 0;
    for(int i : arr[head]){
        if(i != par){
            dfs(i, head, arr, val, bel);
            v += max(bel[i], val[i]);
            bel[head] += val[i];
        }
    }
    val[head] = v;
}
struct node{
    set<int> child;
};
int findSample(int n,int confidence[],int host[],int protocol[]){
    int N = n;
    if(n <= 10){
        vector<vector<int>> arr(N);
        for(int i = 1; i < N; i ++){
            int h = host[i];
            if(protocol[i] == 0){
                arr[i].push_back(h);
                arr[h].push_back(i);
            }
            if(protocol[i] == 1){
                for(int j : arr[h]){
                    arr[i].push_back(j);
                    arr[j].push_back(i);
                }
            }
            if(protocol[i] == 2){
                arr[i].push_back(h);
                arr[h].push_back(i);
                for(int j : arr[h]){
                    arr[i].push_back(j);
                    arr[j].push_back(i);
                }
            }
        }
        int m = 0;
        set<int> brr;
        getmax(0, N, arr, m, brr, 0, confidence);
        return m;
    }
    else{
        if(count(protocol, protocol + n, 1) == n-1){
            return accumulate(confidence, confidence + n, 0ll);
        }
        if(count(protocol, protocol + n, 2) == n-1){
            return *max_element(confidence, confidence + n);
        }
        if(count(protocol, protocol + n, 0) == n - 1){
            vector<vector<int>> arr(N);
            for(int i = 1; i < N; i ++){
                int h = host[i];
                if(protocol[i] == 0){
                    arr[i].push_back(h);
                    arr[h].push_back(i);
                }
            }
            vector<int> val(N), bel(N);
            for(int i = 0; i < N; i ++){
                bel[i] = confidence[i];
            }
            dfs(0, -1, arr, val, bel);
            return val[0];
        }
        vector<node> arr(N);
        for(int i = 1; i < N; i ++){
            int h = host[i];
            if(protocol[i] == 0){
                arr[i].child.insert(h);
                arr[h].child.insert(i);
            }
            else{
                for(int j : arr[h].child){
                    arr[i].child.insert(j);
                }
            }
        }
        int c = 0;
        for(int i = 0; i < N; i ++){
            int mv = 1e9;
            for(int j = 0; j < N; j ++){
                if(arr[j].child.size()){
                    mv = min(mv, (int)arr[j].child.size());
                }
            }
            for(int j = 0; j < N; j ++){
                if(arr[i].child.size() != mv)continue;
                for(int k : arr[j].child){
                    arr[k].child.clear();
                }
                arr[j].child.clear();
                c++;
                break;
            }
        }
        return c;
    }
}

Compilation message

friend.cpp: In function 'int findSample(int, int*, int*, int*)':
friend.cpp:113:40: warning: comparison of integer expressions of different signedness: 'std::set<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
  113 |                 if(arr[i].child.size() != mv)continue;
      |                    ~~~~~~~~~~~~~~~~~~~~^~~~~

Subtask #1
11.0 / 11.0

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

Subtask #2
8.0 / 8.0

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

Subtask #3
8.0 / 8.0

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

Subtask #4
0 / 19.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 0 ms 212 KB Output is correct
5 Incorrect 2 ms 340 KB Output isn't correct
6 Halted 0 ms 0 KB -

Subtask #5
0 / 23.0

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

Subtask #6
0 / 31.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 0 ms 212 KB Output is correct
5 Correct 0 ms 212 KB Output is correct
6 Correct 0 ms 212 KB Output is correct
7 Correct 0 ms 212 KB Output is correct
8 Correct 0 ms 212 KB Output is correct
9 Correct 0 ms 212 KB Output is correct
10 Correct 0 ms 212 KB Output is correct
11 Correct 0 ms 212 KB Output is correct
12 Execution timed out 1081 ms 15060 KB Time limit exceeded
13 Halted 0 ms 0 KB -