結果

問題 No.655 E869120 and Good Triangles
コンテスト
ユーザー vwxyz
提出日時 2022-06-29 05:09:10
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,799 bytes
コンパイル時間 260 ms
コンパイル使用メモリ 13,184 KB
実行使用メモリ 113,920 KB
最終ジャッジ日時 2024-11-22 01:26:27
合計ジャッジ時間 72,310 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 TLE * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from collections import deque
readline=sys.stdin.readline

N,K,P=map(int,readline().split())
inf=1<<30
A=[[inf]*(i+1) for i in range(N)]
queue=deque([])
for _ in range(K):
    x,y=map(int,readline().split())
    x-=1;y-=1
    queue.append((x,y))
    A[x][y]=0
while queue:
    x,y=queue.popleft()
    if 0<=y-1<=x-1<N and A[x][y]+1<A[x-1][y-1]:
        A[x-1][y-1]=A[x][y]+1
        queue.append((x-1,y-1))
    if 0<=y<=x-1<N and A[x][y]+1<A[x-1][y]:
        A[x-1][y]=A[x][y]+1
        queue.append((x-1,y))
    if 0<=y-1<=x<N and A[x][y]+1<A[x][y-1]:
        A[x][y-1]=A[x][y]+1
        queue.append((x,y-1))
    if 0<=y+1<=x<N and A[x][y]+1<A[x][y+1]:
        A[x][y+1]=A[x][y]+1
        queue.append((x,y+1))
    if 0<=y<=x+1<N and A[x][y]+1<A[x+1][y]:
        A[x+1][y]=A[x][y]+1
        queue.append((x+1,y))
    if 0<=y+1<=x+1<N and A[x][y]+1<A[x+1][y+1]:
        A[x+1][y+1]=A[x][y]+1
        queue.append((x+1,y+1))
AL=[[A[i][j] for j in range(i+1)] for i in range(N)]
AR=[[A[i][j] for j in range(i+1)] for i in range(N)]
for i in range(1,N):
    for j in range(i):
        AL[i][j]+=AL[i-1][j]
for i in range(1,N):
    for j in range(1,i+1):
        AR[i][j]+=AR[i-1][j-1]
for i in range(N):
    A[i]=[0]+A[i]
    for j in range(1,i+2):
        A[i][j]+=A[i][j-1]
ans=0
dp=[(0,0)]
for i in range(N):
    prev=dp
    dp=[(0,0)]*(i+2)
    for j in range(i+1):
        ii,s=prev[j]
        while ii<N and s<P:
            s+=A[ii][j-i+ii+1]-A[ii][j]
            ii+=1
        prev[j]=(ii,s)
        if i!=N-1:
            dp[j]=max(dp[j],(ii,s-(AR[ii-1][j-i+ii-1]-(AR[i-1][j-1] if i and j else 0))))
            dp[j+1]=max(dp[j+1],(ii,s-(AL[ii-1][j]-(AL[i-1][j] if i and i!=j else 0))))
    for j in range(i+1):
        if prev[j][1]>=P:
            ans+=N-prev[j][0]+1
print(ans)
0