Submission #362007

#	Time	Username	Problem	Language	Result	Execution time	Memory
362007		lukameladze	Easter Eggs (info1cup17_eastereggs)	C++14	100 / 100	22 ms	572 KiB

eastereggs

# include <bits/stdc++.h>
#include "grader.h"
#define f first
#define s second
#define pb push_back
using namespace std;
long long le,ri, mid;
vector <int> v2;
vector <long long> v1;
vector <long long> vvv[1005];
int ans;
void dfs(int a, int p) {
     v1.pb(a);
     for (int i=0; i<(int)vvv[a].size(); i++) {
          if (vvv[a][i]!=p) {
               dfs(vvv[a][i],a);
          }
     }
}
int findEgg(int n, vector <pair <int, int > > vv)
{
     for (int i=1; i<=515; i++) {
          vvv[i].clear();
     }
     v1.clear();
     v2.clear();
         for (int i=0; i<(int)vv.size(); i++) {
          vvv[vv[i].f].pb(vv[i].s);
          vvv[vv[i].s].pb(vv[i].f);
     }
     dfs(1,0);
     le=0;
     ri=(int)v1.size()-2;
     ans=-1;
     while (le<=ri) {
          mid=(le+ri)/2;
          v2.clear();
          for (int i=0; i<=mid; i++) {
               v2.pb(v1[i]);
          }
          if (query(v2)==1) {
               ans=mid;
               ri=mid-1;
          }
          else le=mid+1;
     }
     if (ans!=-1)
     return v1[ans];
     else return v1[(int)v1.size()-1];
}

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