結果
問題 | No.2774 Wake up Record 2 |
ユーザー |
![]() |
提出日時 | 2024-06-07 21:26:44 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 438 ms / 2,000 ms |
コード長 | 1,518 bytes |
コンパイル時間 | 381 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 94,976 KB |
最終ジャッジ日時 | 2024-12-26 06:57:24 |
合計ジャッジ時間 | 3,931 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 16 |
ソースコード
import ioimport sysfrom collections import defaultdict, deque, Counterfrom itertools import permutations, combinations, accumulatefrom heapq import heappush, heappopfrom bisect import bisect_right, bisect_leftfrom math import gcdimport math_INPUT = """\68 31 3 5 6 4 8 9 1010 101 2 3 2 3 2 3 2 1 110 5199178881 932079191 906976560 427573994 961937646 979063146 995030172 852645442 662699738 313858222"""def input():return sys.stdin.readline()[:-1]def solve(test):N,K=map(int,input().split())A=list(map(int,input().split()))B=sorted([(A[i],i) for i in range(N)])C=sorted([(B[i][1],0 if i<K else 1) for i in range(N)])ans=[]for i in range(1,N):if C[i][1]==1 and C[i-1][1]==0:ans.append(C[i][0]+1)print(len(ans))print(*ans)def random_input():from random import randint,shuffleN=randint(1,10)M=randint(1,N)A=list(range(1,M+1))+[randint(1,M) for _ in range(N-M)]shuffle(A)return (" ".join(map(str, [N,M]))+"\n"+" ".join(map(str, A))+"\n")*3def simple_solve():return []def main(test):if test==0:solve(0)elif test==1:sys.stdin = io.StringIO(_INPUT)case_no=int(input())for _ in range(case_no):solve(0)else:for i in range(1000):sys.stdin = io.StringIO(random_input())x=solve(1)y=simple_solve()if x!=y:print(i,x,y)print(*[line for line in sys.stdin],sep='')break#0:提出用、1:与えられたテスト用、2:ストレステスト用main(0)