Coloring

from heapq import *
from collections import defaultdict

ns = defaultdict(lambda: set())

n,m = input().split()
n = int(n)
m = int(m)

ds = [0] * n


for i in range(m):
    u,v = input().split()
    u = int(u)-1
    v = int(v)-1
    ns[u].add(v)
    ns[v].add(u)
    ds[u] += 1
    ds[v] += 1

pal = [set(range(6)) for i in range(n)]
ds = [(d,i) for i, d in enumerate(ds)]

heapify(ds)

cols = [None] * n

def firstcol(taken):
    for i in range(6):
        if i not in taken:
            return i

dsdict = [len(ns[i]) for i in range(n)]

while ds:
    d,i = heappop(ds)
    if cols[i] is None:
        #ncols = {cols[nb] for nb in ns[i]}
        #cols[i] = firstcol(ncols)
        #print(i, pal[i], dsdict[i], cols[i])
        cols[i] = list(pal[i])[0]
        c = cols[i]
        for nb in ns[i]:
            dsdict[nb] -= 1
            ns[nb].remove(i)
            if c in pal[nb]:
                pal[nb].remove(c)
            heappush(ds, (dsdict[nb], nb))

print(" ".join(str(x+1) for x in cols))
    

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 42, in <module>
    cols[i] = list(pal[i])[0]
IndexError: list index out of range

Test 12

Verdict: TIME LIMIT EXCEEDED

Test 13

Verdict: TIME LIMIT EXCEEDED

Test 14

Verdict: TIME LIMIT EXCEEDED

Test 15

Verdict: ACCEPTED

Test 16

Verdict: ACCEPTED

Test 17

Verdict: TIME LIMIT EXCEEDED

Test 18

Verdict: ACCEPTED

Test 19

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 42, in <module>
    cols[i] = list(pal[i])[0]
IndexError: list index out of range

Test 20

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 42, in <module>
    cols[i] = list(pal[i])[0]
IndexError: list index out of range

Test 21

Verdict: ACCEPTED

Test 22

Verdict: ACCEPTED

Test 23

Verdict: ACCEPTED

Test 24

Verdict: ACCEPTED

Test 25

Verdict: ACCEPTED

Test 26

Verdict: ACCEPTED

Test 27

Verdict: TIME LIMIT EXCEEDED