Submission #347851

#	Time	Username	Problem	Language	Result	Execution time	Memory
347851		beksultan04	Tropical Garden (IOI11_garden)	C++14	49 / 100	116 ms	29100 KiB

garden

#include "garden.h"
#include "gardenlib.h"
//#include "grader.cpp"
#include <bits/stdc++.h>
using namespace std;
#define lol long long
#define pii pair<int,int>
#define OK puts("OK");
#define NO puts("NO");
#define YES puts("YES");
#define fr first
#define sc second
#define ret return
#define scanl(a) scanf("%lld",&a);
#define scanll(a,b) scanf("%lld %lld",&a, &b);
#define scanlll(a,b,c) scanf("%lld %lld %lld",&a,&b,&c);
#define scan1(a) scanf("%d",&a);
#define scan2(a,b) scanf("%d %d",&a, &b);
#define scan3(a,b,c) scanf("%d %d %d",&a,&b,&c);
#define all(s) s.begin(),s.end()
#define allr(s) s.rbegin()Ñ,s.rend()
#define pb push_back
#define sz(v) (int)v.size()
#define endi puts("");
#define eps 1e-12
const int INF=1e9+7;
int n,p,ans=0,f,k,up[300001][31];
vector <int> v[200001],g;

void count_routes(int N, int M, int P, int R[][2], int Q, int G[])
{
    int i,j;
    for (i=0;i<M;++i){
        if (v[R[i][0]].size() < 2)
            v[R[i][0]].pb(R[i][1]);

        if (v[R[i][1]].size() < 2)
            v[R[i][1]].pb(R[i][0]);

    }

    for (int pr=0;pr<N;++pr){
        if (v[pr].size() == 2){
            {
                int nd = v[pr][0];
                if (v[nd].size() == 2){
                    if (v[nd][0] == pr){
                        up[pr][0] = nd+N;
                    }
                    else {
                        up[pr][0] = nd;
                    }
                }
            }
            {
                int nd = v[pr][1];
                if (v[nd].size() == 2){
                    if (v[nd][0] == pr){
                        up[pr+N][0] = nd+N;
                    }
                    else {
                        up[pr+N][0] = nd;
                    }
                }
            }
        }
        else {
            int nd = v[pr][0];
            if (v[nd].size() == 2){
                if (v[nd][0] == pr){
                    up[pr][0] = nd+N;
                }
                else {
                    up[pr][0] = nd;
                }
            }
        }
    }

    for (i=1;i<31;++i){
        for (j=0;j<2*N;++j){
            up[j][i] = up[up[j][i-1]][i-1];
        }
    }

    for (j=0;j<Q;++j){
        for (i=0;i<N;++i){
            int v = i;
            k = G[j];
            for (int pw = 0;pw < 31;++pw){
                if (k & (1<<pw)){
                    v = up[v][pw];
                }
            }
            if (v == P || v == P+N)ans++;
        }
        answer(ans);
        ans=0;
    }
}

• Subtask 1: 49 / 49
• Subtask 2: 0 / 20
• Subtask 3: 0 / 31