結果
問題 | No.1028 闇討ち |
ユーザー | Kodai |
提出日時 | 2020-04-23 20:31:21 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
RE
|
実行時間 | - |
コード長 | 1,006 bytes |
コンパイル時間 | 84 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 92,432 KB |
最終ジャッジ日時 | 2024-04-21 22:30:19 |
合計ジャッジ時間 | 33,267 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | RE | - |
testcase_01 | AC | 533 ms
44,384 KB |
testcase_02 | AC | 529 ms
44,000 KB |
testcase_03 | AC | 533 ms
44,516 KB |
testcase_04 | AC | 532 ms
44,512 KB |
testcase_05 | AC | 538 ms
43,880 KB |
testcase_06 | AC | 540 ms
44,644 KB |
testcase_07 | AC | 530 ms
44,256 KB |
testcase_08 | AC | 530 ms
44,264 KB |
testcase_09 | AC | 600 ms
44,252 KB |
testcase_10 | AC | 631 ms
44,128 KB |
testcase_11 | AC | 973 ms
51,688 KB |
testcase_12 | TLE | - |
testcase_13 | TLE | - |
testcase_14 | AC | 1,534 ms
65,212 KB |
testcase_15 | AC | 848 ms
47,200 KB |
testcase_16 | TLE | - |
testcase_17 | TLE | - |
testcase_18 | TLE | - |
testcase_19 | TLE | - |
testcase_20 | TLE | - |
testcase_21 | TLE | - |
ソースコード
import numpy as np N = int(input()) A_list = [list(map(int, input().split())) for i in range(N)] upper_count = np.array([0 for i in range(N)]) for i in range(N): for j in range(N): upper_count[A_list[i][j] - 1] += i if i > j else j add_list = np.array([np.array([0 for i in range(N)]) for j in range(N - 1)]) minus_list = np.array([np.array([0 for i in range(N)]) for j in range(N - 1)]) for i in range(0, N - 1): for j in range(i + 1): add_list[i][A_list[i - j][j] - 1] += 1 minus_list[-(i + 1)][A_list[N - 1 - (i - j)][j] - 1] += 1 if i != 0: add_list[i] = add_list[i - 1] + add_list[i] minus_list[-(i + 1)] = minus_list[-i] + minus_list[-(i + 1)] for i in range(len(minus_list) - 1): add_list[i + 1] += add_list[i] minus_list[i + 1] += minus_list[i] result_list = add_list - minus_list min_list = np.amin(result_list, axis=0) min_list = np.vectorize(lambda a: a if a < 0 else 0)(min_list) print(np.sum(upper_count) + np.sum(min_list))