結果

問題 No.335 門松宝くじ
コンテスト
ユーザー mkawa2
提出日時 2020-02-28 18:57:35
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 381 ms / 2,000 ms
コード長 1,983 bytes
コンパイル時間 224 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,136 KB
最終ジャッジ日時 2024-10-13 16:31:16
合計ジャッジ時間 2,416 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

sys.setrecursionlimit(10 ** 6)
int1 = lambda x: int(x) - 1
p2D = lambda x: print(*x, sep="\n")
def II(): return int(sys.stdin.readline())
def MI(): return map(int, sys.stdin.readline().split())
def LI(): return list(map(int, sys.stdin.readline().split()))
def LLI(rows_number): return [LI() for _ in range(rows_number)]
def SI(): return sys.stdin.readline()[:-1]

class SparseTable:
    def __init__(self, aa):
        w = len(aa)
        h = w.bit_length()
        table = [aa] + [[-1] * w for _ in range(h - 1)]
        tablei1 = table[0]
        for i in range(1, h):
            tablei = table[i]
            for j in range(w-(1<<i)+1):
                rj = j + (1 << (i - 1))
                tablei[j] = min(tablei1[j], tablei1[rj])
            tablei1 = tablei
        self.table = table

    def min(self, l, r):
        i = (r - l).bit_length() - 1
        tablei = self.table[i]
        Lmin = tablei[l]
        Rmin = tablei[r - (1 << i)]
        if Lmin < Rmin: Rmin = Lmin
        return Rmin

def main():
    inf=10**9
    n,m=MI()
    ss=[]
    for _ in range(m):
        ee=LI()
        st=SparseTable(ee)
        lmx = [0] * n
        rmx = [0] * n
        mx=-inf
        for i,e in enumerate(ee):
            lmx[i]=mx
            if e>mx:mx=e
        mx=-inf
        for i,e in enumerate(ee[::-1]):
            rmx[n-1-i]=mx
            if e>mx:mx=e
        #print(lmx)
        #print(lmn)
        #print(rmx)
        #print(rmn)
        s=0
        for j in range(n):
            for i in range(j):
                a,b=ee[i],ee[j]
                if a<b:
                    if lmx[j]>b:s+=lmx[j]
                    elif lmx[i]>a or st.min(j,n)<b or st.min(i,j)<a:s+=b
                else:
                    if rmx[i]>a:s+=rmx[i]
                    elif rmx[j]>b or st.min(0,i+1)<a or st.min(i,j)<b:s+=a
        ss.append(s)
    #print(ss)
    mx=max(ss)
    for i in range(m):
        if ss[i]==mx:
            print(i)
            break

main()
0