# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
278707 | jairRS | 늑대인간 (IOI18_werewolf) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "werewolf.h"
#include <set>
#include <vector>
#include <algorithm>
using namespace std;
set<int> adj[3004];
int memo[3004][2];
int dp(int curcity, int endcity, int l, int r, vector<bool> visited, int form = 0)
{
if (form == 0 & curcity < l)
return 0;
if (form == 1 & curcity > r)
return 0;
if (curcity == endcity && form == 1)
return 1;
if (memo[curcity][form] != -1)
return memo[curcity][form];
visited[curcity] = true;
int res = 0;
for (int adjcity : adj[curcity])
{
if (visited[adjcity])
{
continue;
}
res += dp(adjcity, endcity, l, r, visited, form);
}
if (form != 1)
{
vector<bool> nvisited(visited.size());
res += dp(curcity, endcity, l, r, nvisited, 1);
}
memo[curcity][form] = res;
return res;
}
std::vector<int> check_validity(int N, std::vector<int> X, std::vector<int> Y,
std::vector<int> S, std::vector<int> E,
std::vector<int> L, std::vector<int> R)
{
int Q = S.size();
int M = X.size();
for (int i = 0; i < M; i++)
{
adj[X[i]].insert(Y[i]);
adj[Y[i]].insert(X[i]);
}
std::vector<int> A(Q);
for (int i = 0; i < Q; i++)
{
vector<bool> nvisited(N);
memset(memo, -1, sizeof(memo));
int ans = dp(S[i], E[i], L[i], R[i], nvisited);
ans = min(ans, 1);
A[i] = ans;
}
return A;
}