# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
838791 | ErJ | 열쇠 (IOI21_keys) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <vector>
#include <unordered_map>
#include <cmath>
#include <algorithm>
#include <iostream>
#include <queue>
using namespace std;
typedef long long ll;
vector<vector<pair<int, int>>> edges; //kam, key
vector<int> p, r;
vector<vector<int>> kv;//potrebuji klic na hranu vedouci do vrcholu
vector<bool> was;
int DFS(int u) {
int ans = 0;
queue<int> q;
q.push(u);
was[u] = true;
kv[r[u]][0] = -2;
while (!q.empty()) {
int v = q.front();
q.pop();
for (int i = 0; i < edges[v].size(); i++) {
//cout << i << endl;
int w = edges[v][i].first;
int kw = edges[v][i].second;
if (kv[kw][0] == -2) {
if (!was[w]) {
was[w] = true;
q.push(w);
ans++;
if (kv[r[w]][0] != -2) {
kv[r[w]][0] = -2;
for (int i = 1; i < kv[r[w]].size(); i++) {
int x = kv[r[w]][i];
if (!was[x]) {
was[x] = true;
q.push(x);
ans++;
}
}
}
}
}
else {
if(!was[w]) kv[kw].push_back(w);
}
}
}
return ans;
}