CSES - Aalto Competitive Programming 2024 - wk2 - Mon - Results
Submission details
Task:Building Teams
Sender:peik-e
Submission time:2024-09-09 17:26:15 +0300
Language:C++ (C++20)
Status:READY
Result:ACCEPTED
Test results
testverdicttime
#1ACCEPTED0.00 sdetails
#2ACCEPTED0.00 sdetails
#3ACCEPTED0.00 sdetails
#4ACCEPTED0.00 sdetails
#5ACCEPTED0.00 sdetails
#6ACCEPTED0.35 sdetails
#7ACCEPTED0.35 sdetails
#8ACCEPTED0.35 sdetails
#9ACCEPTED0.35 sdetails
#10ACCEPTED0.21 sdetails
#11ACCEPTED0.00 sdetails
#12ACCEPTED0.00 sdetails

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