結果

問題 No.2735 Demarcation
ユーザー kusirakusirakusirakusira
提出日時 2024-04-01 22:55:00
言語 PyPy3
(7.3.15)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,891 bytes
コンパイル時間 269 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 268,672 KB
最終ジャッジ日時 2024-10-11 11:42:08
合計ジャッジ時間 22,131 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
58,112 KB
testcase_01 AC 39 ms
52,608 KB
testcase_02 AC 38 ms
52,352 KB
testcase_03 AC 39 ms
52,480 KB
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 TLE -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 TLE -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 TLE -
testcase_21 AC 424 ms
234,804 KB
testcase_22 AC 307 ms
268,672 KB
testcase_23 TLE -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

# 入力
n = int(input())
X = list(map(int,input().split()))
q = int(input())

INF = 3*10**18

# 尺取り法の関数
def add(x, cnt):
    if(x not in cnt):
        cnt[x] = 0
    cnt[x] += 1

def remove(x, cnt):
    cnt[x] -= 1
    if(cnt[x] == 0):
        del cnt[x]

def judge(x, cnt, k):
    add(x, cnt)
    flag = len(cnt) <= k
    remove(x, cnt)
    return flag

def dp(k):
    if(k == 0): return 0
    l = 0
    r = 0
    # 遷移先
    to = [n] * n
    # 要素数カウント
    cnt = {}

    ans = -1
    # 尺取り法
    while r < n:
        if(judge(A[r], cnt, k)):
            add(A[r], cnt)
            r += 1
        else:
            remove(A[l], cnt)
            l += 1
        to[l] = r

    dp = [0] * (n+1)
    dp[0] = 1
    dp[1] = -1

    for i in range(n):
        # 累積
        if(i != 0):
            dp[i] += dp[i-1]
            if(dp[i] > s):
                return INF
        # 遷移
        dp[i+1] += dp[i]
        if(to[i]+1 < n+1):
            dp[to[i]+1] -= dp[i]

    ans = dp[-1] + dp[-2]
    return ans

# 同じ要素が隣り合っているものの個数累積和
CX = [0]
for i in range(n-1):
    CX.append(CX[-1] + int(X[i] == X[i+1]))
CX.append(CX[-1])

for _ in range(q):
    l,r,s = map(int,input().split())
    l -= 1
    r -= 1

    if(l-r+1 >= 62):
        # f(1)の値
        c = CX[r+1]-CX[l+1]
        if(c >= 62):
            print(0)
        else:
            f1 = CX[r+1]-CX[l+1]
            if(f1 <= s):
                print(f1)
            else:
                print(0)
    else:
        A = X[l:r+1]
        # めぐる式二分探索
        ok = 0
        ng = 70
        n = len(A)
        while abs(ok-ng)>1:
            mid = (ok+ng)//2
            if(dp(mid)<=s):
                ok = mid
            else:
                ng = mid
        if(ok >= 65):
            print(-1)
        else:
            print(dp(ok))
0