結果

問題 No.2095 High Rise
ユーザー lam6er
提出日時 2025-04-16 00:31:56
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 260 ms / 2,000 ms
コード長 996 bytes
コンパイル時間 509 ms
コンパイル使用メモリ 81,484 KB
実行使用メモリ 171,792 KB
最終ジャッジ日時 2025-04-16 00:33:50
合計ジャッジ時間 3,868 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main():
    import sys
    input = sys.stdin.read().split()
    idx = 0
    N = int(input[idx])
    idx += 1
    M = int(input[idx])
    idx += 1
    if N == 1:
        print(0)
        return
    A = []
    for _ in range(N):
        row = list(map(int, input[idx:idx+M]))
        idx += M
        A.append(row)
    
    # Initialize DP
    dp_prev = [0] * M
    for j in range(M):
        dp_prev[j] = A[0][j] + A[1][j]
    
    for i in range(N-2):
        current = [float('inf')] * M
        min_prev = min(dp_prev)
        for j in range(M):
            # Option 1: extend the interval in j
            if i + 2 < N:
                option1 = dp_prev[j] + A[i+2][j]
            else:
                option1 = float('inf')
            # Option 2: start new interval in j
            option2 = min_prev + A[i+1][j] + (A[i+2][j] if i+2 < N else 0)
            current[j] = min(option1, option2)
        dp_prev = current
    
    print(min(dp_prev))

if __name__ == '__main__':
    main()
0