CSES - Aalto Competitive Programming 2024 - wk2 - Mon - Results
Submission details
Task:Building Teams
Sender:Meriem
Submission time:2024-09-09 16:36:55 +0300
Language:CPython3
Status:READY
Result:
Test results
testverdicttime
#1ACCEPTED0.02 sdetails
#2ACCEPTED0.02 sdetails
#3ACCEPTED0.02 sdetails
#4ACCEPTED0.02 sdetails
#50.02 sdetails
#60.68 sdetails
#70.67 sdetails
#80.68 sdetails
#90.67 sdetails
#100.78 sdetails
#110.02 sdetails
#120.02 sdetails

Code

def task1(n, relations, team):
    for i in range(1, n + 1):
        if team[i] == 0:
            if not dfs(i, relations, team):
                print("IMPOSSIBLE")
    team = team[1:]
    print(" ".join(map(str, team)))


def dfs(i, relations, team):
    if team[i] == 0:
        team[i] = 1
    if team[i] == 1:
        friend_team = 2
    else:
        friend_team = 1
    for friend in relations[i]:
        if team[friend] == 0:
            team[friend] = friend_team
            if not dfs(friend, relations, team):
                return False
        elif team[friend] != friend_team:
            return False
    return True


if __name__ == "__main__":
    n, m = map(int, input().split())
    relations = [[] for _ in range(n + 1)]
    team = [0] * (n + 1)

    for i in range(m):
        u, v = map(int, input().split())
        relations[u].append(v)
        relations[v].append(u)
    task1(n, relations, team)

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:

input
10 20
3 5
8 10
9 10
1 8
...

correct output
IMPOSSIBLE

user output
IMPOSSIBLE
IMPOSSIBLE
1 1 2 2 1 1 2 2 2 1

Test 6

Verdict:

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
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 36, in <module>
    task1(n, relations, team)
  File "/box/input/code.py", line 4, in task1
    if not dfs(i, relations, team):
  File "/box/input/code.py", line 20, in dfs
    if not dfs(friend, relations, team):
  File "/box/input/code.py", line 20, in dfs
    if not dfs(friend, relations, team):
  File "/box/input/code.py", line 20, in dfs
    if not dfs(friend, relations, team):
  [Previous line repeated 994 more times]
  File "/box/input/code.py", line 11, in dfs
    if team[i] == 0:
RecursionError: maximum recursion depth exceeded in comparison

Test 7

Verdict:

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
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 36, in <module>
    task1(n, relations, team)
  File "/box/input/code.py", line 4, in task1
    if not dfs(i, relations, team):
  File "/box/input/code.py", line 20, in dfs
    if not dfs(friend, relations, team):
  File "/box/input/code.py", line 20, in dfs
    if not dfs(friend, relations, team):
  File "/box/input/code.py", line 20, in dfs
    if not dfs(friend, relations, team):
  [Previous line repeated 994 more times]
  File "/box/input/code.py", line 11, in dfs
    if team[i] == 0:
RecursionError: maximum recursion depth exceeded in comparison

Test 8

Verdict:

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
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 36, in <module>
    task1(n, relations, team)
  File "/box/input/code.py", line 4, in task1
    if not dfs(i, relations, team):
  File "/box/input/code.py", line 20, in dfs
    if not dfs(friend, relations, team):
  File "/box/input/code.py", line 20, in dfs
    if not dfs(friend, relations, team):
  File "/box/input/code.py", line 20, in dfs
    if not dfs(friend, relations, team):
  [Previous line repeated 994 more times]
  File "/box/input/code.py", line 11, in dfs
    if team[i] == 0:
RecursionError: maximum recursion depth exceeded in comparison

Test 9

Verdict:

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
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 36, in <module>
    task1(n, relations, team)
  File "/box/input/code.py", line 4, in task1
    if not dfs(i, relations, team):
  File "/box/input/code.py", line 20, in dfs
    if not dfs(friend, relations, team):
  File "/box/input/code.py", line 20, in dfs
    if not dfs(friend, relations, team):
  File "/box/input/code.py", line 20, in dfs
    if not dfs(friend, relations, team):
  [Previous line repeated 994 more times]
  File "/box/input/code.py", line 11, in dfs
    if team[i] == 0:
RecursionError: maximum recursion depth exceeded in comparison

Test 10

Verdict:

input
100000 200000
38942 96755
70049 82663
7746 72732
87819 99029
...

correct output
IMPOSSIBLE

user output
IMPOSSIBLE
IMPOSSIBLE
IMPOSSIBLE
IMPOSSIBLE
IMPOSSIBLE
...

Test 11

Verdict:

input
5 4
1 2
3 4
4 5
5 3

correct output
IMPOSSIBLE

user output
IMPOSSIBLE
1 2 1 2 1

Test 12

Verdict:

input
4 5
1 2
1 4
2 3
2 4
...

correct output
IMPOSSIBLE

user output
IMPOSSIBLE
1 2 1 2