CSES - Datatähti 2021 loppu - Results
Submission details
Task:Kolmijako
Sender:Microwave Abuser
Submission time:2021-01-23 17:12:37 +0200
Language:PyPy3
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED22
#2ACCEPTED78
Test results
testverdicttimegroup
#1ACCEPTED0.05 s1, 2details
#2ACCEPTED0.05 s1, 2details
#3ACCEPTED0.05 s1, 2details
#4ACCEPTED0.05 s1, 2details
#5ACCEPTED0.05 s1, 2details
#6ACCEPTED0.05 s1, 2details
#7ACCEPTED0.05 s1, 2details
#8ACCEPTED0.05 s1, 2details
#9ACCEPTED0.05 s2details
#10ACCEPTED0.05 s2details
#11ACCEPTED0.05 s2details
#12ACCEPTED0.05 s2details
#13ACCEPTED0.05 s2details
#14ACCEPTED0.05 s2details
#15ACCEPTED0.05 s2details

Code

def main():
    try:
        n = int(input())
        if n == 3:
            print("1 2 3")
            return
        result = []
        result.append(0)
        sum = 0
        for i in range(n):
            sum += i+1
            result.append(0)
        groupsum = sum/3

        cursum = 0
        index = n
        while cursum != groupsum:
            if cursum + index <= groupsum:
                result[index] = 1
                cursum += index
            index -= 1

        cursum = 0
        index = n
        while cursum != groupsum:
            if cursum + index <= groupsum and result[index] == 0:
                result[index] = 2
                cursum += index
            index -= 1

        onegroup = result[1]
        resultstr = ""
        if onegroup == 2:
            result[1] = 0
            for i in range(n):
                if result[i+1] == 0:
                    resultstr += "3 "
                elif result[i+1] == 1:
                    resultstr += "2 "
                else:
                    resultstr += "1 "
        elif onegroup == 1:
            result[1] = 2
            for i in range(n):
                if result[i+1] == 0:
                    resultstr += "2 "
                elif result[i+1] == 1:
                    resultstr += "1 "
                else:
                    resultstr += "3 "
        else:
            result[1] = 2
            for i in range(n):
                if result[i+1] == 0:
                    resultstr += "1 "
                elif result[i+1] == 1:
                    resultstr += "2 "
                else:
                    resultstr += "3 "
    except:
        print("IMPOSSIBLE")
        return
    print(resultstr[:-1])


main()

Test details

Test 1

Group: 1, 2

Verdict: ACCEPTED

input
3

correct output
1 2 3 

user output
1 2 3

Test 2

Group: 1, 2

Verdict: ACCEPTED

input
4

correct output
IMPOSSIBLE

user output
IMPOSSIBLE

Test 3

Group: 1, 2

Verdict: ACCEPTED

input
5

correct output
1 3 1 3 2 

user output
3 3 3 1 2

Test 4

Group: 1, 2

Verdict: ACCEPTED

input
6

correct output
1 3 2 2 1 3 

user output
3 3 2 2 3 1

Test 5

Group: 1, 2

Verdict: ACCEPTED

input
7

correct output
IMPOSSIBLE

user output
IMPOSSIBLE

Test 6

Group: 1, 2

Verdict: ACCEPTED

input
8

correct output
2 3 1 2 3 3 2 1 

user output
3 1 1 2 3 1 3 2

Test 7

Group: 1, 2

Verdict: ACCEPTED

input
9

correct output
1 2 3 1 2 3 3 2 1 

user output
3 1 1 1 1 2 3 3 2

Test 8

Group: 1, 2

Verdict: ACCEPTED

input
10

correct output
IMPOSSIBLE

user output
IMPOSSIBLE

Test 9

Group: 2

Verdict: ACCEPTED

input
42

correct output
1 3 2 2 1 3 1 2 3 3 2 1 1 2 3 ...

user output
3 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

Test 10

Group: 2

Verdict: ACCEPTED

input
95

correct output
1 3 1 3 2 1 2 3 3 2 1 1 2 3 3 ...

user output
3 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

Test 11

Group: 2

Verdict: ACCEPTED

input
96

correct output
1 3 2 2 1 3 1 2 3 3 2 1 1 2 3 ...

user output
3 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

Test 12

Group: 2

Verdict: ACCEPTED

input
97

correct output
IMPOSSIBLE

user output
IMPOSSIBLE

Test 13

Group: 2

Verdict: ACCEPTED

input
98

correct output
2 3 1 2 3 3 2 1 1 2 3 3 2 1 1 ...

user output
3 1 1 1 1 2 1 1 1 1 1 1 1 1 1 ...

Test 14

Group: 2

Verdict: ACCEPTED

input
99

correct output
1 2 3 1 2 3 3 2 1 1 2 3 3 2 1 ...

user output
3 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

Test 15

Group: 2

Verdict: ACCEPTED

input
100

correct output
IMPOSSIBLE

user output
IMPOSSIBLE