結果

問題 No.2735 Demarcation
ユーザー yosupotyosupot
提出日時 2024-04-19 22:59:34
言語 Rust
(1.77.0 + proconio)
結果
TLE  
実行時間 -
コード長 2,302 bytes
コンパイル時間 14,033 ms
コンパイル使用メモリ 404,864 KB
実行使用メモリ 292,736 KB
最終ジャッジ日時 2024-10-11 17:04:06
合計ジャッジ時間 22,678 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
10,624 KB
testcase_01 AC 1 ms
5,248 KB
testcase_02 AC 1 ms
5,248 KB
testcase_03 AC 1 ms
5,248 KB
testcase_04 AC 228 ms
292,736 KB
testcase_05 AC 47 ms
28,092 KB
testcase_06 AC 281 ms
237,824 KB
testcase_07 AC 399 ms
278,016 KB
testcase_08 AC 336 ms
269,540 KB
testcase_09 AC 138 ms
105,856 KB
testcase_10 AC 192 ms
146,944 KB
testcase_11 AC 66 ms
62,080 KB
testcase_12 AC 38 ms
20,992 KB
testcase_13 AC 82 ms
38,124 KB
testcase_14 AC 308 ms
204,544 KB
testcase_15 AC 79 ms
28,288 KB
testcase_16 AC 297 ms
239,744 KB
testcase_17 AC 210 ms
141,056 KB
testcase_18 AC 88 ms
63,488 KB
testcase_19 AC 372 ms
275,840 KB
testcase_20 AC 385 ms
257,228 KB
testcase_21 AC 170 ms
197,248 KB
testcase_22 AC 238 ms
284,932 KB
testcase_23 TLE -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: variable does not need to be mutable
  --> src/main.rs:64:17
   |
64 |             let mut k = (lw + up) / 2;
   |                 ----^
   |                 |
   |                 help: remove this `mut`
   |
   = note: `#[warn(unused_mut)]` on by default

ソースコード

diff # 

use core::cmp::{max, min};
use proconio::input;

fn main() {
    input! {
        n: i32,
        x: [i32; n],
        q: i32,
        ques: [(i32, i32, i64); q],
    }

    let mut count_same = vec![0; n as usize];
    for i in 0..(n - 1) {
        count_same[(i + 1) as usize] = count_same[i as usize]
            + if x[i as usize] == x[(i + 1) as usize] {
                1
            } else {
                0
            }
    }

    const MAX_K: i32 = 70;
    let mut range = vec![[n; (MAX_K + 1) as usize]; n as usize];
    let mut vis = vec![-1; (n + 1) as usize];
    for i in 0..n {
        let mut vis_count = 0;
        for j in (max(0, i - MAX_K + 1)..i + 1).rev() {
            if vis[x[j as usize] as usize] != i {
                vis_count += 1;
                vis[x[j as usize] as usize] = i;
            }
            range[i as usize][vis_count] = j;
        }
        for k in (vis_count as i32)..MAX_K + 1 {
            range[i as usize][k as usize] = 0;
        }
    }

    for (l, r, s) in ques {
        let l = l - 1;

        let dist = r - l;

        if dist <= 62 && 1i64 << (dist - 1) <= s {
            println!("-1");
            continue;
        }

        let same = count_same[(r - 1) as usize] - count_same[l as usize];

        if 61 <= same || s < (1i64 << same) {
            println!("0");
            continue;
        }

        const INF: i128 = 1i128 << 62;
        let mut dp = vec![0i128; (dist + 1) as usize];
        let mut dp_sum = vec![0i128; (dist + 1 + 1) as usize];

        let mut lw = 1;
        let mut up = MAX_K;

        while up - lw > 1 {
            let mut k = (lw + up) / 2;

            dp[0] = 1;
            dp_sum[0] = 0;
            dp_sum[1] = 1;
            for i in l..r {
                dp[(i + 1 - l) as usize] = min(
                    INF,
                    dp_sum[(i + 1 - l) as usize]
                        - dp_sum[(max(l, range[i as usize][k as usize]) - l) as usize],
                );
                dp_sum[(i + 1 - l + 1) as usize] =
                    dp_sum[(i + 1 - l) as usize] + dp[(i + 1 - l) as usize];
            }

            if dp[dist as usize] <= s as i128 {
                lw = k;
            } else {
                up = k;
            }
        }
        println!("{}", lw);
    }
}
0