Submission #529684

#	Time	Username	Problem	Language	Result	Execution time	Memory
529684		ohohorz	Easter Eggs (info1cup17_eastereggs)	C++14	0 / 100	2 ms	336 KiB

eastereggs

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

using namespace std;

const int MAXN = 520;
set<int> g[MAXN];
int n, sub[MAXN];
vector<int> cenTree[MAXN], what[MAXN];

int dfs(int u,int p){
    sub[u] = 1;
    for(auto v:g[u]) if(v != p){
        sub[u] += dfs(v, u);
    }
    return sub[u];

}
int centroid(int u, int p, int mx){
    for(auto v:g[u]) if(v != p){
        if(sub[v] > (mx / 2)) return centroid(v, u, mx);
    }
    return u;
}
void make(int u, int p){
    int mx = dfs(u, p);
    int c = centroid(u, p, mx);
    if(p == 0) p = c;
    else{
        cenTree[p].push_back(c);
        cenTree[c].push_back(p);
    }
    
    vector<int> temp(g[c].begin(), g[c].end());
    for(auto v:temp){
        g[c].erase(v);
        g[v].erase(c);
        make(v, c);
    }
}

void dfs2(int u,int p){
    what[u].push_back(u);
    for(auto v:cenTree[u]) if(v != p){
        dfs2(v, u);
        for(auto x:what[v]) what[u].push_back(v);
    }
}

int get(int u,int p){
    for(auto v:g[u]) if(v != p){
        int q = query(what[v]);
        if(q == 1){
            return get(v, u);
        }
    }
    return u;
}


int findEgg (int N, vector < pair < int, int > > bridges){   
    n = N;
    for(int i = 0;i < N-1;i++){
        int u = bridges[i].first;
        int v = bridges[i].second;
        g[u].insert(v);
        g[v].insert(u);
    }

    make(1, 0);
    // for(int i =1;i <= n;i++){
    //     cout << i << " " << par[i] << "\n";
    // }
    


    return get(1, 0);
}

Compilation message (stderr)

eastereggs.cpp: In function 'void dfs2(int, int)':
eastereggs.cpp:46:18: warning: unused variable 'x' [-Wunused-variable]
   46 |         for(auto x:what[v]) what[u].push_back(v);
      |                  ^

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