結果
| 問題 | 
                            No.942 プレゼント配り
                             | 
                    
| コンテスト | |
| ユーザー | 
                             | 
                    
| 提出日時 | 2019-12-05 00:32:34 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                WA
                                 
                             
                            
                         | 
                    
| 実行時間 | - | 
| コード長 | 654 bytes | 
| コンパイル時間 | 767 ms | 
| コンパイル使用メモリ | 82,000 KB | 
| 実行使用メモリ | 110,708 KB | 
| 最終ジャッジ日時 | 2024-12-14 17:00:46 | 
| 合計ジャッジ時間 | 3,431 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge1 / judge2 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 2 | 
| other | AC * 14 WA * 4 | 
ソースコード
import sys
N, K = map(int, input().split())
if N * (N + 1) // 2 % K != 0:
  print('No')
  sys.exit()
if N // K % 2 == 1:
  m = N // K
  print('Yes')
  c = []
  l = N // K // K * K * K
  for i in range(l, (l + N) // 2):
    c.append(i + 1)
    c.append(N - (i - l))
  for i in range(K):
    a = []
    for j in range(N // K // K * K):
      a.append(K * j + (i + j) % K + 1)
    u=(N-l)//K
    a += c[u*i:u*(i+1)]
    print(*a)
elif N // K % 2 == 0:
  print('Yes')
  a = []
  for i in range(N // 2):
    a.append(i)
    a.append(N - 1 - i)
  for i in range(K):
    b = []
    for j in range(N // K):
      b.append(a[i * N // K + j] + 1)
    print(*b)