結果

問題 No.165 四角で囲え！
コンテスト
ユーザー mkawa2
提出日時 2020-01-24 16:58:24
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,170 ms / 5,000 ms
コード長 2,236 bytes
コンパイル時間 168 ms
コンパイル使用メモリ 82,100 KB
実行使用メモリ 78,720 KB
最終ジャッジ日時 2024-09-14 03:44:58
合計ジャッジ時間 16,028 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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]

# 座標圧縮すればxyともにN=400までに抑えられる
# 点の数、得点の和ともに、事前に2次元累積和の表を作っておけばO(1)で求められる
# それでもすべての四角形について全探索するとO(N**4)ぐらいでタイムアウト
# そこで上下の辺（つまりy座標）については全探索し、y座標を固定したうえで、
# 左右の辺（x座標）をしゃくとり法で動かしていく
# それでO(N**3)なのでギリ間に合うんじゃね？って感じ
def main():
    n, b = MI()
    # 入力を受けながら座標圧縮
    xx = set()
    yy = set()
    plot = []
    for _ in range(n):
        x, y, p = MI()
        plot.append((x, y, p))
        xx.add(x)
        yy.add(y)
    comx = {x: i for i, x in enumerate(sorted(xx))}
    comy = {y: i for i, y in enumerate(sorted(yy))}
    xn, yn = len(comx), len(comy)
    # 点の数cと得点sの累積和を作ろう
    c = [[0] * (yn + 1) for _ in range(xn + 1)]
    s = [[0] * (yn + 1) for _ in range(xn + 1)]
    for x, y, p in plot:
        x, y = comx[x] + 1, comy[y] + 1
        c[x][y] = 1
        s[x][y] = p
    for x in range(1, xn + 1):
        for y in range(1, yn + 1):
            c[x][y] += c[x][y - 1]
            s[x][y] += s[x][y - 1]
    for y in range(1, yn + 1):
        for x in range(1, xn + 1):
            c[x][y] += c[x - 1][y]
            s[x][y] += s[x - 1][y]
    # p2D(c)
    # p2D(s)
    # yを固定しながらxをしゃくとり
    ans = 0
    for y0 in range(yn):
        for y1 in range(y0 + 1, yn + 1):
            x1 = 0
            for x0 in range(xn):
                while x1 < xn and s[x1 + 1][y1] + s[x0][y0] - s[x0][y1] - s[x1 + 1][y0] <= b: x1 += 1
                cnt = c[x1][y1] + c[x0][y0] - c[x0][y1] - c[x1][y0]
                if cnt > ans: ans = cnt
    print(ans)

main()
0