Task: | Building Teams |
Sender: | peik-e |
Submission time: | 2024-09-09 17:26:15 +0300 |
Language: | C++ (C++20) |
Status: | READY |
Result: | ACCEPTED |
test | verdict | time | |
---|---|---|---|
#1 | ACCEPTED | 0.00 s | details |
#2 | ACCEPTED | 0.00 s | details |
#3 | ACCEPTED | 0.00 s | details |
#4 | ACCEPTED | 0.00 s | details |
#5 | ACCEPTED | 0.00 s | details |
#6 | ACCEPTED | 0.35 s | details |
#7 | ACCEPTED | 0.35 s | details |
#8 | ACCEPTED | 0.35 s | details |
#9 | ACCEPTED | 0.35 s | details |
#10 | ACCEPTED | 0.21 s | details |
#11 | ACCEPTED | 0.00 s | details |
#12 | ACCEPTED | 0.00 s | details |
Code
#include <algorithm> #include <iostream> #include <queue> #include <set> #include <stack> #include <vector> using namespace std; const int A = 0; const int B = 1; bool dfs(int p, int set_team, vector<int> &teams, const vector<vector<int>> &edges, set<int> &to_visit) { // imp. if (teams[p] == !set_team) return false; // visited // if (to_visit.find(p) == to_visit.end()) // return teams[p] == set_team; to_visit.erase(p); teams[p] = set_team; // recurse for (const auto &f : edges[p]) { if (to_visit.find(f) != to_visit.end()) { if (!dfs(f, !set_team, teams, edges, to_visit)) { return false; } } else if (teams[f] == set_team) { return false; } } return true; } int main(void) { int n, m; cin >> n >> m; vector<int> teams(n); vector<vector<int>> edges(n); set<int> to_visit; for (int i = 0; i < n; ++i) { edges.push_back({}); to_visit.insert(i); } fill(teams.begin(), teams.end(), -1); for (int j = 0; j < m; ++j) { int a, b; cin >> a >> b; edges[a - 1].push_back(b - 1); edges[b - 1].push_back(a - 1); } while (!to_visit.empty()) { if (!dfs(*to_visit.begin(), 0, teams, edges, to_visit)) { cout << "IMPOSSIBLE" << endl; return 0; } // for (int i = 0; i < n; ++i) { // cout << teams[i] + 1; // } // cout << endl; } for (int i = 0; i < n; ++i) { cout << teams[i] + 1 << ' '; } cout << endl; return 0; }
Test details
Test 1
Verdict: ACCEPTED
input |
---|
10 20 3 4 8 10 3 7 1 8 ... |
correct output |
---|
1 1 1 2 2 1 2 2 2 1 |
user output |
---|
1 1 1 2 2 1 2 2 2 1 |
Test 2
Verdict: ACCEPTED
input |
---|
10 20 1 3 8 10 2 4 6 8 ... |
correct output |
---|
1 1 2 2 1 1 1 2 1 1 |
user output |
---|
1 1 2 2 1 1 1 2 1 1 |
Test 3
Verdict: ACCEPTED
input |
---|
10 20 7 10 3 10 9 10 2 10 ... |
correct output |
---|
1 2 2 1 1 1 2 1 2 1 |
user output |
---|
1 2 2 1 1 1 2 1 2 1 |
Test 4
Verdict: ACCEPTED
input |
---|
10 20 2 4 2 10 7 10 4 6 ... |
correct output |
---|
1 2 1 1 2 2 2 1 2 1 |
user output |
---|
1 2 1 1 2 2 2 1 2 1 |
Test 5
Verdict: ACCEPTED
input |
---|
10 20 3 5 8 10 9 10 1 8 ... |
correct output |
---|
IMPOSSIBLE |
user output |
---|
IMPOSSIBLE |
Test 6
Verdict: ACCEPTED
input |
---|
100000 200000 47355 96505 90709 92058 735 80715 91802 94265 ... |
correct output |
---|
1 2 2 1 2 1 1 1 2 2 1 2 1 1 1 ... |
user output |
---|
1 2 2 1 2 1 1 1 2 2 1 2 1 1 1 ... Truncated |
Test 7
Verdict: ACCEPTED
input |
---|
100000 200000 59991 95794 95150 96051 78453 94730 90411 95523 ... |
correct output |
---|
1 1 1 2 2 1 1 2 1 2 1 2 2 2 1 ... |
user output |
---|
1 1 1 2 2 1 1 2 1 2 1 2 2 2 1 ... Truncated |
Test 8
Verdict: ACCEPTED
input |
---|
100000 200000 89827 96402 65137 86792 80965 94708 19479 48078 ... |
correct output |
---|
1 2 1 1 2 1 2 2 2 1 2 1 1 2 1 ... |
user output |
---|
1 2 1 1 2 1 2 2 2 1 2 1 1 2 1 ... Truncated |
Test 9
Verdict: ACCEPTED
input |
---|
100000 200000 72952 83723 66197 70052 2949 52160 55753 95651 ... |
correct output |
---|
1 1 2 2 2 1 1 2 2 2 2 2 1 2 1 ... |
user output |
---|
1 1 2 2 2 1 1 2 2 2 2 2 1 2 1 ... Truncated |
Test 10
Verdict: ACCEPTED
input |
---|
100000 200000 38942 96755 70049 82663 7746 72732 87819 99029 ... |
correct output |
---|
IMPOSSIBLE |
user output |
---|
IMPOSSIBLE |
Test 11
Verdict: ACCEPTED
input |
---|
5 4 1 2 3 4 4 5 5 3 |
correct output |
---|
IMPOSSIBLE |
user output |
---|
IMPOSSIBLE |
Test 12
Verdict: ACCEPTED
input |
---|
4 5 1 2 1 4 2 3 2 4 ... |
correct output |
---|
IMPOSSIBLE |
user output |
---|
IMPOSSIBLE |