結果

問題 No.165 四角で囲え！
ユーザー mkawa2mkawa2
提出日時 2020-01-24 16:53:56
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
実行時間 -
コード長 2,236 bytes
コンパイル時間 207 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 29,632 KB
最終ジャッジ日時 2024-09-14 03:44:10
合計ジャッジ時間 12,734 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 TLE -
testcase_01 -- -
testcase_02 -- -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
権限があれば一括ダウンロードができます

ソースコード

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 = 1
            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