結果

問題 No.1031 いたずら好きなお姉ちゃん
ユーザー MNGOFMNGOF
提出日時 2020-04-26 21:41:41
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
WA  
実行時間 -
コード長 1,791 bytes
コンパイル時間 130 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 49,928 KB
最終ジャッジ日時 2024-04-29 09:14:49
合計ジャッジ時間 63,579 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 519 ms
44,168 KB
testcase_01 AC 509 ms
44,296 KB
testcase_02 AC 520 ms
44,420 KB
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 RE -
testcase_35 RE -
testcase_36 RE -
testcase_37 RE -
testcase_38 RE -
testcase_39 RE -
testcase_40 RE -
testcase_41 RE -
testcase_42 RE -
testcase_43 RE -
testcase_44 RE -
testcase_45 RE -
testcase_46 RE -
testcase_47 RE -
testcase_48 RE -
testcase_49 RE -
testcase_50 RE -
testcase_51 RE -
testcase_52 RE -
testcase_53 RE -
testcase_54 AC 510 ms
44,168 KB
testcase_55 AC 518 ms
44,688 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import numpy as np


def solve_left(area, selected_ind):
    if area.shape[0] <= 1:
        return 0
    max_ind = np.argmax(area)
    min_ind = max_ind
    selected_ind[max_ind: selected_ind.shape[0]] = 0
    for i in range(max_ind+1, area.shape[0]):
        if area[min_ind] > area[i]:
            min_ind = i
            selected_ind[i] = 1
    count = np.sum(selected_ind)
    count += solve_left(area[0:max_ind], selected_ind[0:max_ind])
    count += solve_right(area[max_ind+1:area.shape[0]], selected_ind[max_ind+1:selected_ind.shape[0]])
    return count


def solve_right(area, selected_ind):
    if area.shape[0] <= 1:
        return 0
    max_ind = np.argmax(area)
    min_ind = max_ind
    selected_ind[0: max_ind + 1] = 0
    for i in reversed(range(max_ind)):
        if area[min_ind] > area[i]:
            min_ind = i
            selected_ind[i] = 1
    count = np.sum(selected_ind)
    count += solve_left(area[0:max_ind], selected_ind[0:max_ind])
    count += solve_right(area[max_ind+1:area.shape[0]], selected_ind[max_ind+1:selected_ind.shape[0]])
    return count


def solve(area):
    selected_ind = area*0
    max_ind = np.argmax(area)
    min_ind = max_ind
    for i in reversed(range(max_ind + 0)):
        if area[min_ind] > area[i]:
            min_ind = i
            selected_ind[i] = 1
    min_ind = max_ind
    for i in range(max_ind+1, area.shape[0]):
        if area[min_ind] > area[i]:
            min_ind = i
            selected_ind[i] = 1
    count = np.sum(selected_ind)
    count += solve_left(area[0:max_ind], selected_ind[0:max_ind])
    count += solve_right(area[max_ind+1:area.shape[0]], selected_ind[max_ind+1:selected_ind.shape[0]])
    return count

N = int(input())
input_np = np.array(input().split()).astype(np.int32)

print(solve(input_np))
0