結果

問題 No.165 四角で囲え！
ユーザー mkawa2mkawa2
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 751 ms
77,696 KB
testcase_01 AC 39 ms
52,736 KB
testcase_02 AC 39 ms
52,608 KB
testcase_03 AC 39 ms
52,352 KB
testcase_04 AC 40 ms
52,864 KB
testcase_05 AC 1,018 ms
77,608 KB
testcase_06 AC 315 ms
76,476 KB
testcase_07 AC 40 ms
52,608 KB
testcase_08 AC 40 ms
52,268 KB
testcase_09 AC 40 ms
52,352 KB
testcase_10 AC 40 ms
52,096 KB
testcase_11 AC 472 ms
77,052 KB
testcase_12 AC 224 ms
76,524 KB
testcase_13 AC 255 ms
76,432 KB
testcase_14 AC 169 ms
76,356 KB
testcase_15 AC 179 ms
76,244 KB
testcase_16 AC 1,471 ms
78,456 KB
testcase_17 AC 1,300 ms
78,528 KB
testcase_18 AC 2,170 ms
78,720 KB
testcase_19 AC 1,378 ms
77,996 KB
testcase_20 AC 1,328 ms
78,336 KB
testcase_21 AC 1,302 ms
77,952 KB
testcase_22 AC 1,311 ms
78,232 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