結果

問題 No.165 四角で囲え！
ユーザー mkawa2mkawa2
提出日時 2020-01-24 16:58:24
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,260 ms / 5,000 ms
コード長 2,236 bytes
コンパイル時間 334 ms
コンパイル使用メモリ 87,328 KB
実行使用メモリ 80,220 KB
最終ジャッジ日時 2023-10-12 04:12:28
合計ジャッジ時間 17,791 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 797 ms
78,976 KB
testcase_01 AC 75 ms
71,432 KB
testcase_02 AC 75 ms
71,592 KB
testcase_03 AC 74 ms
71,412 KB
testcase_04 AC 74 ms
71,120 KB
testcase_05 AC 1,115 ms
79,392 KB
testcase_06 AC 346 ms
77,808 KB
testcase_07 AC 74 ms
71,388 KB
testcase_08 AC 75 ms
71,440 KB
testcase_09 AC 75 ms
71,436 KB
testcase_10 AC 74 ms
71,580 KB
testcase_11 AC 496 ms
78,704 KB
testcase_12 AC 254 ms
77,676 KB
testcase_13 AC 286 ms
77,668 KB
testcase_14 AC 199 ms
77,468 KB
testcase_15 AC 206 ms
77,572 KB
testcase_16 AC 1,529 ms
79,580 KB
testcase_17 AC 1,412 ms
79,664 KB
testcase_18 AC 2,260 ms
80,220 KB
testcase_19 AC 1,481 ms
79,620 KB
testcase_20 AC 1,370 ms
79,716 KB
testcase_21 AC 1,362 ms
79,644 KB
testcase_22 AC 1,354 ms
79,716 KB
権限があれば一括ダウンロードができます

ソースコード

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]

# 座標圧縮すれば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