結果

問題 No.2735 Demarcation
ユーザー yosupotyosupot
提出日時 2024-04-19 23:18:39
言語 Rust
(1.77.0)
結果
AC  
実行時間 148 ms / 1,500 ms
コード長 2,427 bytes
コンパイル時間 1,570 ms
コンパイル使用メモリ 183,520 KB
実行使用メモリ 20,968 KB
最終ジャッジ日時 2024-04-19 23:18:45
合計ジャッジ時間 5,281 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 0 ms
5,248 KB
testcase_01 AC 0 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 1 ms
5,376 KB
testcase_04 AC 32 ms
11,716 KB
testcase_05 AC 21 ms
5,376 KB
testcase_06 AC 58 ms
16,640 KB
testcase_07 AC 128 ms
19,584 KB
testcase_08 AC 85 ms
15,592 KB
testcase_09 AC 47 ms
8,704 KB
testcase_10 AC 60 ms
11,136 KB
testcase_11 AC 15 ms
5,632 KB
testcase_12 AC 19 ms
5,376 KB
testcase_13 AC 41 ms
5,376 KB
testcase_14 AC 103 ms
14,976 KB
testcase_15 AC 49 ms
5,376 KB
testcase_16 AC 65 ms
16,768 KB
testcase_17 AC 78 ms
9,088 KB
testcase_18 AC 35 ms
5,376 KB
testcase_19 AC 102 ms
19,272 KB
testcase_20 AC 125 ms
18,304 KB
testcase_21 AC 42 ms
11,136 KB
testcase_22 AC 50 ms
16,108 KB
testcase_23 AC 31 ms
7,168 KB
testcase_24 AC 68 ms
10,752 KB
testcase_25 AC 90 ms
16,244 KB
testcase_26 AC 41 ms
12,544 KB
testcase_27 AC 44 ms
11,136 KB
testcase_28 AC 148 ms
20,968 KB
testcase_29 AC 69 ms
17,180 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: unused import: `max`
 --> main.rs:1:17
  |
1 | use core::cmp::{max, min};
  |                 ^^^
  |
  = note: `#[warn(unused_imports)]` on by default

warning: unnecessary parentheses around function argument
  --> main.rs:43:47
   |
43 |         let mut dp_sum = vec![0i128; min(300, (dist + 1 + 1)) as usize];
   |                                               ^            ^
   |
   = note: `#[warn(unused_parens)]` on by default
help: remove these parentheses
   |
43 -         let mut dp_sum = vec![0i128; min(300, (dist + 1 + 1)) as usize];
43 +         let mut dp_sum = vec![0i128; min(300, dist + 1 + 1) as usize];
   |

warning: 2 warnings emitted

ソースコード

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 freq = vec![0; (n + 1) as usize];

    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;
        }

        let mut dp_sum = vec![0i128; min(300, (dist + 1 + 1)) as usize];

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

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

            dp_sum[0] = 0;
            dp_sum[1] = 1;

            let mut left = l;
            let mut vis_count = 0;

            let mut over = false;            
            for i in l..r {
                if freq[x[i as usize] as usize] == 0 {
                    vis_count += 1;
                }
                freq[x[i as usize] as usize] += 1;
                while vis_count == k + 1 {
                    freq[x[left as usize] as usize] -= 1;
                    if freq[x[left as usize] as usize] == 0 {
                        vis_count -= 1;
                    }
                    left += 1;
                }

                let dp = dp_sum[(i + 1 - l) as usize] - dp_sum[(left - l) as usize];
                if dp > s as i128 {
                    for j in l..i + 1 {
                        freq[x[j as usize] as usize] = 0;
                    }
                    over = true;
                    break;
                }
                dp_sum[(i + 1 - l + 1) as usize] =
                    dp_sum[(i + 1 - l) as usize] + dp;
            }
            
            if !over {
                for i in l..r {
                    freq[x[i as usize] as usize] = 0;
                }
                lw = k;
            } else {
                up = k;
            }
        }
        println!("{}", lw);
    }
}
0