結果
問題 | 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 io import sys from collections import defaultdict, deque, Counter from itertools import permutations, combinations, accumulate from heapq import heappush, heappop from bisect import bisect_right, bisect_left from math import gcd import math _INPUT = """\ 6 8 3 1 3 5 6 4 8 9 10 10 10 1 2 3 2 3 2 3 2 1 1 10 5 199178881 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,shuffle N=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")*3 def 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)