-
Notifications
You must be signed in to change notification settings - Fork 814
/
Copy path1170. Safari Park (25).cpp
38 lines (37 loc) · 1.08 KB
/
1170. Safari Park (25).cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
#include <iostream>
#include <map>
#include <vector>
using namespace std;
int N, R, K, M, flag, E[505][505];
int main() {
cin >> N >> R >> K;
for (int i = 1, u, v; i <= R; i++) {
cin >> u >> v;
E[u][v] = E[v][u] = 1;
}
for (cin >> M; M; M--) {
map<int, vector<int>> A;
for (int i = 1, a; i <= N; i++) {
cin >> a;
A[a].push_back(i);
}
if(A.size() > K) cout << "Error: Too many species.\n";
else if (A.size() < K) cout << "Error: Too few species.\n";
else {
flag = 1;
for (auto it : A) {
for (int i = 0; i < it.second.size(); i++) {
for (int j = i + 1; j < it.second.size(); j++) {
if(E[it.second[i]][it.second[j]]) {
flag = 0;
break;
}
}
}
}
if (flag) cout << "Yes\n";
else cout << "No\n";
}
}
return 0;
}