Submission #920606

#	Time	Username	Problem	Language	Result	Execution time	Memory
920606		DON_F	Easter Eggs (info1cup17_eastereggs)	C++14	0 / 100	1 ms	492 KiB

eastereggs

#include <bits/stdc++.h>
#include "grader.h"

using namespace std;
using ll = long long;
#define F first
#define S second

vector < vector < int > > g;
set < int > cur, sp;
vector < int > all, vis;
ll p;

void dfs(int u){
   if ((int) sp.size() == p){
      return;
   }
   vis[u] = 1;
   all.push_back(u);
   if (cur.count(u)){
     sp.insert(u);
   }
   for (auto &i : g[u]){
     if (!vis[i])
        dfs(i);
   }
}

int findEgg (int n, vector < pair < int, int > > z){
    for (int i = 1; i <= n; ++i){
        cur.insert(i);
    }
    g = vector < vector < int > > (n + 1);
    for (int i = 0; i < n - 1; ++i){
        g[z[i].F].push_back(z[i].S);
    }
    while (cur.size() > 1){
        p = cur.size() / 2;
        for (int i = 1; i <= n; ++i){
            vis = vector < int > (n + 1);
            dfs(i);
            if ((int) sp.size() == p){
                break;
            }
            sp.clear();
            all.clear();
        }
        ll tmp = query(all);
        if (tmp){
           cur.clear();
           for (auto &i : sp){
              cur.insert(i);
           }
        }else{
           for (auto &i : sp){
             cur.erase(i);
           }
        }
        sp.clear();
        all.clear();
    }
    if (cur.size())
       return *cur.begin();
    else return -1;
}

• Subtask 1: 0 / 40
• Subtask 2: 0 / 40
• Subtask 3: 0 / 20