제출 #439890

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
439890		dutch	열대 식물원 (Tropical Garden) (IOI11_garden)	C++17	100 / 100	1740 ms	25544 KiB

garden

#include <bits/stdc++.h>
using namespace std;
#include "garden.h"
#include "gardenlib.h"

const int LIM = 1.5e5, INF = 2e9;

int n, cnt[LIM], c, d[2][LIM*2], p, r[2] = {INF, INF};
vector<int> g[LIM*2];

void dfs(int u, int dist){
	if(!d[c][u]) return void(r[c] = dist);
	d[c][u] = dist;
	for(int v : g[u]) dfs(v, dist + 1);
}
 
void count_routes(int N, int M, int P, int R[][2], int Q, int G[]){
	n = N, p = P;
	for(int i=0; i<M; ++i){
		for(int j=0; j<2; ++j){
			int u = R[i][j], v = R[i][!j];
			if(cnt[u] == 0) g[v + (!cnt[v])*n].push_back(u);
			if(cnt[u] == 1) g[v + (!cnt[v])*n].push_back(u+n);
		}
		for(int j=0; j<2; ++j) ++cnt[R[i][j]];
	}
	for(int u=0; u<n; ++u) if(cnt[u] < 2){
		for(int v : g[u+n]) g[u].push_back(v);
		g[u+n].clear();
	}
	for(c=0; c<2; ++c) fill(d[c], d[c]+2*n, INF);
	for(c=0; c<2; ++c) dfs(p+c*n, 0);

	int e[2][n];
	for(int u=0; u<n; ++u)
		for(int j=0; j<2; ++j)
			e[j][u] = d[j][u] % r[j];
 
	for(int i=0; i<Q; ++i){
		int res = 0, x[2] = {G[i] % r[0], G[i] % r[1]};
		for(int u=0; u<n; ++u)
			for(int j=0; j<2; ++j)
				if(d[j][u] <= G[i] && x[j] == e[j][u]) ++res;
		answer(res);
	}
}

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