Submission #231986

#	Time	Username	Problem	Language	Result	Execution time	Memory
231986		DodgeBallMan	Tropical Garden (IOI11_garden)	C++14	0 / 100	11 ms	9856 KiB

garden

#include <bits/stdc++.h>
#include "gardenlib.h"
#include "garden.h"

using namespace std;

const int N = 3e5 + 10;
int chk[N], d[N], to[N], cyc;
vector<int> vec, g[N];

void dfs( int u ) {
    //printf("U %d\n",u);
    chk[u] = 1, vec.emplace_back( u );
    for( int v : g[u] ) {
        if( !chk[v] ) d[v] = d[u] + 1, dfs( v );
        else if( chk[v] == 1 ) for( int i = (int)vec.size() - 1 ; i >= 0 ; i-- ) {
            cyc++;
            if( vec[i] == v ) break;
        }
    }
    chk[u] = 2, vec.pop_back();
}

void count_routes( int n, int m, int p, int r[][2], int q, int G[] ) {
    for( int i = 0 ; i < m ; i++ ) {
        if( g[r[i][0]].size() < 2 ) g[r[i][0]].emplace_back( r[i][1] );
        if( g[r[i][1]].size() < 2 ) g[r[i][1]].emplace_back( r[i][0] );
    }
    for( int i = 0 ; i < n ; i++ ) for( int j = 0 ; j < g[i].size() ; j++ ) {
        int v = g[i][j], now = i + j*n;
        if( g[v][0] == i && g[v].size() > 1 ) {
            to[now] = v + n;
            if( g[i].size() == 1 ) to[i+n] = v + n;
        }
        else {
            to[now] = v;
            if( g[i].size() == 1 ) to[i+n] = v + n;
        }
        //printf("u v : %d %d\n",i,v);
    }
    /*for( int i = 0 ; i < 2*n ; i++ ) printf("%d ",to[i]);
    printf("\n");*/
    for( int i = 0 ; i < 2*n ; i++ ) g[i].clear();
    for( int i = 0 ; i < 2*n ; i++ ) g[to[i]].emplace_back( i );
    /*for( int i = 0 ; i < 2*n ; i++ ) {
        printf("%d : ",i);
        for( int v : g[i] ) printf("%d ",v);
        printf("\n");
    }
    printf("\n");*/
    vector<int> ans( q, 0 );
    vec.clear(), memset( d, 0, sizeof d ), cyc = 0, memset( chk, 0, sizeof chk );
    dfs( p );
    //printf("%d\n",cyc);
    for( int i = 0 ; i < q ; i++ ) {
        for( int j = 0 ; j < n ; j++ ) {
            if( d[j] > G[i] ) continue ;
            if( cyc ) ans[i] += ( ( G[i] - d[j] ) % cyc == 0 );
            else ans[i] += ( d[j] == G[i] );
        }
    }
    vec.clear(), memset( d, 0, sizeof d ), cyc = 0, memset( chk, 0, sizeof chk );
    dfs( p+n );
    //printf("%d\n",cyc);
    for( int i = 0 ; i < q ; i++ ) {
        for( int j = 0 ; j < n ; j++ ) {
            if( d[j] > G[i] ) continue ;
            if( cyc ) ans[i] += ( ( G[i] - d[j] ) % cyc == 0 );
            else ans[i] += ( d[j] == G[i] );
        }
        answer( ans[i] );   
    }
    //return ans;
}

/*int main()
{
    int n, m, p, q, r[100][2], G[100];
    scanf("%d %d %d %d",&n,&m,&p,&q);
    for( int i = 0 ; i < m ; i++ ) {
        scanf("%d %d",&r[i][0],&r[i][1]);
    }
    for( int i = 0 ; i < q ; i++ ) scanf("%d",&G[i]);
    vector<int> ans = count_routes( n, m, p, r, q, G );
    for( int i : ans ) printf("%d ",i);
}*/

Compilation message (stderr)

garden.cpp: In function 'void count_routes(int, int, int, int (*)[2], int, int*)':
garden.cpp:29:55: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for( int i = 0 ; i < n ; i++ ) for( int j = 0 ; j < g[i].size() ; j++ ) {
                                                     ~~^~~~~~~~~~~~~

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