제출 #1325627

#제출 시각아이디문제언어결과실행 시간메모리
1325627adiatEaster Eggs (info1cup17_eastereggs)C++20
100 / 100
44 ms600 KiB
#include<bits/stdc++.h>
#include"grader.h"
using namespace std;


vector<vector<int>> adj;
map<int,int> vis, searched;
set<int> s;

void dfs(int n, int &m, vector<int> &v)
{
    vis[n]=1;
    v.push_back(n);
    if(s.find(n)==s.end())
    {
        searched[n]++;
        m--;
        if(m==0) return;
    }
    for(int i: adj[n])
    {
        if(!vis[i])
            dfs(i, m, v);
        if(m==0) return;
    }
}

int findEgg(int N, vector < pair < int, int > > bridges)
{
    adj.assign(N+1, vector<int> ());
    s.clear();
    searched.clear();
    for(auto [u,v]: bridges)
    {
        adj[u].push_back(v);
        adj[v].push_back(u);
    }
    // int available=N;
    int logN=log2(N)+1;
    for(int i=0; i<logN; i++)
    {
        int available=N-s.size();
        if(available==1)
        {
            for(int i=1; i<=N; i++)
            {
                if(s.find(i)==s.end())
                    return i;
            }
        }
        int m=available/2;
        vis.clear();
        vector<int> v;
        dfs(1, m, v);
        int q=query(v);
        if(q)
        {
            for(int i=1; i<=N; i++)
            {
                if(find(v.begin(),v.end(),i)==v.end()) s.insert(i);
            }
        }
        else 
        {
            for(int i: v)
            {
                s.insert(i);
            }
        }
        // for(int i: s) cout<<i<<" ";
        // cout<<endl;
    }
    for(int i=1; i<=N; i++)
    {
        if(s.find(i)==s.end())
            return i;
    }
    return -1;
}

// static int N, X, cntQ;
// static vector < int > v[1009];

// int query (vector < int > h)
// {
//     cntQ ++;
//     int ap[1009];
//     if (h.empty ()) return 0;
//     for (int i=1; i<=N; i++)
//         ap[i] = 0;
//     for (auto it = h.begin (); it != h.end (); it ++)
//         ap[*it] = 1;
//     queue < int > cc;
//     cc.push (h[0]), ap[h[0]] = 2;
//     while (!cc.empty ())
//     {
//         int nod = cc.front ();
//         cc.pop ();
//         for (auto it = v[nod].begin (); it != v[nod].end (); it ++)
//             if (ap[*it] == 1)
//                 ap[*it] = 2, cc.push (*it);
//     }
//     for (int i=1; i<=N; i++)
//         if (ap[i] == 1) return -1;
//     for (auto it = h.begin (); it != h.end (); it ++)
//         if (*it == X) return 1;
//     return 0;
// }

// int main ()
// {
// // freopen ("input", "r", stdin);
// //freopen ("output", "w", stdout);

// scanf ("%d", &N);
// int Queries;
// vector < pair < int, int > > param;
// for (int i=1; i<N; i++)
// {
//     int x, y;
//     scanf ("%d %d", &x, &y);
//     v[x].push_back (y);
//     v[y].push_back (x);
//     param.push_back ({x, y});
// }
// scanf ("%d", &Queries);
// while (Queries --)
// {
//     scanf ("%d", &X), cntQ = 0;
//     int Y = findEgg (N, param);
//     if (X != Y)
//     {
//         printf ("WA %d instead of %d\n", Y, X);
//         return 0;
//     }
//     printf ("OK %d\n", cntQ);
// }
// return 0;
// }
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...