CSES - Datatähti 2021 loppu

Submission details
Task:	Kolmijako
Sender:	Microwave Abuser
Submission time:	2021-01-23 17:12:37 +0200
Language:	Python3 (PyPy3)
Status:	READY
Result:	100

Feedback
group	verdict	score
#1	ACCEPTED	22
#2	ACCEPTED	78

Test results
test	verdict	time	group
#1	ACCEPTED	0.05 s	1, 2	details
#2	ACCEPTED	0.05 s	1, 2	details
#3	ACCEPTED	0.05 s	1, 2	details
#4	ACCEPTED	0.05 s	1, 2	details
#5	ACCEPTED	0.05 s	1, 2	details
#6	ACCEPTED	0.05 s	1, 2	details
#7	ACCEPTED	0.05 s	1, 2	details
#8	ACCEPTED	0.05 s	1, 2	details
#9	ACCEPTED	0.05 s	2	details
#10	ACCEPTED	0.05 s	2	details
#11	ACCEPTED	0.05 s	2	details
#12	ACCEPTED	0.05 s	2	details
#13	ACCEPTED	0.05 s	2	details
#14	ACCEPTED	0.05 s	2	details
#15	ACCEPTED	0.05 s	2	details

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` view save

correct output
`1 2 3` view save

user output
`1 2 3` view save

Test 2

Group: 1, 2

Verdict: ACCEPTED

input
`4` view save

correct output
`IMPOSSIBLE` view save

user output
`IMPOSSIBLE` view save

Test 3

Group: 1, 2

Verdict: ACCEPTED

input
`5` view save

correct output
`1 3 1 3 2` view save

user output
`3 3 3 1 2` view save

Test 4

Group: 1, 2

Verdict: ACCEPTED

input
`6` view save

correct output
`1 3 2 2 1 3` view save

user output
`3 3 2 2 3 1` view save

Test 5

Group: 1, 2

Verdict: ACCEPTED

input
`7` view save

correct output
`IMPOSSIBLE` view save

user output
`IMPOSSIBLE` view save

Test 6

Group: 1, 2

Verdict: ACCEPTED

input
`8` view save

correct output
`2 3 1 2 3 3 2 1` view save

user output
`3 1 1 2 3 1 3 2` view save

Test 7

Group: 1, 2

Verdict: ACCEPTED

input
`9` view save

correct output
`1 2 3 1 2 3 3 2 1` view save

user output
`3 1 1 1 1 2 3 3 2` view save

Test 8

Group: 1, 2

Verdict: ACCEPTED

input
`10` view save

correct output
`IMPOSSIBLE` view save

user output
`IMPOSSIBLE` view save

Test 9

Group: 2

Verdict: ACCEPTED

input
`42` view save

correct output
`1 3 2 2 1 3 1 2 3 3 2 1 1 2 3 ...` view save

user output
`3 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...` view save

Test 10

Group: 2

Verdict: ACCEPTED

input
`95` view save

correct output
`1 3 1 3 2 1 2 3 3 2 1 1 2 3 3 ...` view save

user output
`3 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...` Truncated

Test 11

Group: 2

Verdict: ACCEPTED

input
`96` view save

correct output
`1 3 2 2 1 3 1 2 3 3 2 1 1 2 3 ...` view save

user output
`3 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...` Truncated

Test 12

Group: 2

Verdict: ACCEPTED

input
`97` view save

correct output
`IMPOSSIBLE` view save

user output
`IMPOSSIBLE` view save

Test 13

Group: 2

Verdict: ACCEPTED

input
`98` view save

correct output
`2 3 1 2 3 3 2 1 1 2 3 3 2 1 1 ...` view save

user output
`3 1 1 1 1 2 1 1 1 1 1 1 1 1 1 ...` Truncated

Test 14

Group: 2

Verdict: ACCEPTED

input
`99` view save

correct output
`1 2 3 1 2 3 3 2 1 1 2 3 3 2 1 ...` view save

user output
`3 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...` Truncated

Test 15

Group: 2

Verdict: ACCEPTED

input
`100` view save

correct output
`IMPOSSIBLE` view save

user output
`IMPOSSIBLE` view save

Datatähti 2021 loppu

Kolmijako

Code

Test details

Test 1

Test 2

Test 3

Test 4

Test 5

Test 6

Test 7

Test 8

Test 9

Test 10

Test 11

Test 12

Test 13

Test 14

Test 15

Tasks

Submissions (Microwave Abuser)