結果

問題 No.1170 Never Want to Walk
ユーザー しーある
提出日時 2020-08-15 08:00:14
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 989 bytes
コンパイル時間 181 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 268,560 KB
最終ジャッジ日時 2024-10-10 17:14:55
合計ジャッジ時間 11,814 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30 TLE * 1 -- * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

from bisect import*
def main():
    n,a,b,*x=map(int,open(0).read().split())
    e=[[]]+[[i+i,i-~i]for i in range(1,n)]+[[]for _ in range(n)]
    for i,y in enumerate(x,n):
        l,r=bisect(x,y-b-1)+n,bisect(x,y-a)+n
        while l<r:
            if l&1:
                e[i]+=l,
                l+=1
            if r&1:
                r-=1
                e[i]+=r,
            l>>=1
            r>>=1
        l,r=bisect(x,y+a-1)+n,bisect(x,y+b)+n
        while l<r:
            if l&1:
                e[i]+=l,
                l+=1
            if r&1:
                r-=1
                e[i]+=r,
            l>>=1
            r>>=1
    a=[0]*n
    for s in range(n,n+n):
        if a[s-n]:continue
        c=1
        f={s}
        q=[s]
        for v in q:
            for w in e[v]:
                if w in f:continue
                if w>=n:c+=1
                f|={w}
                q+=w,
        for v in f:
            if v>=n:a[v-n]=c
    print('\n'.join(map(str,a)))
main()
0