# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
900392 | JakobZorz | Tropical Garden (IOI11_garden) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include"garden.h"
#include"gardenlib.h"
#include<iostream>
#include<vector>
using namespace std;
int n; // num nodes
int m; // num edges
int p; // destination
vector<vector<int>>nodes; // lowest node and second lowest node
vector<vector<int>>dist;
void add_conn(int node,int dest){
if(nodes[node][0]==-1){
nodes[node]={dest,dest};
}else if(nodes[node][0]==nodes[node][1]){
nodes[node][1]=dest;
}
}
int dfs(int node,int blocked){
if(dist[node][blocked]!=-1)
return dist[node][blocked];
dist[node][blocked]=1e9;
int blocked2=0;
int nxt=nodes[node][blocked];
if(nodes[nxt][0]==node&&nodes[nxt][1]!=-1)
blocked2=1;
dist[node][blocked]=dfs(nxt,blocked2)+1;
return dist[node][blocked];
}
void count_routes(int N,int M,int P,int R[][2],int Q,int G[]){
n=N;
m=M;
p=P;
nodes=vector<vector<int>>(n,{-1,-1});
dist=vector<vector<int>>(n,{-1,-1});
for(int i=0;i<m;i++){
add_conn(R[i][0],R[i][1]);
add_conn(R[i][1],R[i][0]);
}
dist[p]={0,0};
unordered_map<int,int>mp;
for(int i=0;i<n;i++){
mp[dfs(i,0)]++;
}
for(int i=0;i<Q;i++)
answer(mp[G[i]]);
}