結果

問題 No.165 四角で囲え!
ユーザー titia
提出日時 2023-01-03 01:00:39
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,579 ms / 5,000 ms
コード長 1,120 bytes
コンパイル時間 552 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 80,056 KB
最終ジャッジ日時 2024-11-27 01:34:19
合計ジャッジ時間 20,046 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff #

N,B=map(int,input().split())
P=[list(map(int,input().split())) for i in range(N)]

X=[]
Y=[]

for x,y,_ in P:
    X.append(x)
    Y.append(y)

X=sorted(set(X))
Y=sorted(set(Y))

DX={X[i]:i for i in range(len(X))}
DY={Y[i]:i for i in range(len(Y))}

for i in range(N):
    P[i][0]=DX[P[i][0]]
    P[i][1]=DY[P[i][1]]


MAP=[[0]*(len(Y)+3) for i in range(len(X)+3)]
MAPP=[[0]*(len(Y)+3) for i in range(len(X)+3)]

for x,y,p in P:
    MAP[x][y]+=p
    MAPP[x][y]+=1

for i in range(1,len(X)+2):
    for j in range(len(Y)+2):
        MAP[i][j]+=MAP[i-1][j]
        MAPP[i][j]+=MAPP[i-1][j]

for i in range(len(X)+2):
    for j in range(1,len(Y)+2):
        MAP[i][j]+=MAP[i][j-1]
        MAPP[i][j]+=MAPP[i][j-1]

ANS=0

for i in range(len(X)):
    for j in range(i,len(X)):
        rind=0

        for lind in range(len(Y)):
            while rind<len(Y):
                if MAP[j][rind]-MAP[j][lind-1]-MAP[i-1][rind]+MAP[i-1][lind-1]<=B:
                    ANS=max(ANS,MAPP[j][rind]-MAPP[j][lind-1]-MAPP[i-1][rind]+MAPP[i-1][lind-1])
                    rind+=1
                else:
                    break

print(ANS)
0