結果

問題 No.2680 研究室配属
ユーザー mymelochan
提出日時 2024-03-20 21:08:29
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 131 ms / 2,000 ms
コード長 855 bytes
コンパイル時間 287 ms
コンパイル使用メモリ 82,556 KB
実行使用メモリ 85,120 KB
最終ジャッジ日時 2024-09-30 06:54:19
合計ジャッジ時間 3,308 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

#############################################################

import sys
sys.setrecursionlimit(10**7)

from heapq import heappop,heappush
from collections import deque,defaultdict,Counter
from bisect import bisect_left, bisect_right
from itertools import product,combinations,permutations

ipt = sys.stdin.readline

def iin():
    return int(ipt())
def lmin():
    return list(map(int,ipt().split()))

MOD = 998244353
#############################################################

N,M = lmin()
A = lmin()
T = [lmin() for _ in range(N)]

ans = [-1]*N
R = list(range(N))
TT = [[] for _ in range(N)]
for i in range(2*N):
    if not R:
        break
    NR = []
    
    for r in R:
        tar = T[r][i]
        if len(TT[tar]) == A[tar]:
            NR.append(r)
        else:
            TT[tar].append(r)
            ans[r] = tar
    R = NR

print(*ans)
0