結果

問題 No.1496 不思議な数え上げ
ユーザー 37kt37kt
提出日時 2024-11-26 11:38:28
言語 Rust
(1.77.0 + proconio)
結果
AC  
実行時間 409 ms / 3,000 ms
コード長 1,309 bytes
コンパイル時間 13,062 ms
コンパイル使用メモリ 378,188 KB
実行使用メモリ 15,872 KB
最終ジャッジ日時 2024-11-26 11:38:59
合計ジャッジ時間 29,752 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 1 ms
5,248 KB
testcase_03 AC 409 ms
14,336 KB
testcase_04 AC 347 ms
14,336 KB
testcase_05 AC 351 ms
14,464 KB
testcase_06 AC 307 ms
15,872 KB
testcase_07 AC 302 ms
15,744 KB
testcase_08 AC 302 ms
15,744 KB
testcase_09 AC 357 ms
15,744 KB
testcase_10 AC 351 ms
15,616 KB
testcase_11 AC 339 ms
15,744 KB
testcase_12 AC 341 ms
14,336 KB
testcase_13 AC 287 ms
15,872 KB
testcase_14 AC 289 ms
15,872 KB
testcase_15 AC 327 ms
15,872 KB
testcase_16 AC 321 ms
15,744 KB
testcase_17 AC 385 ms
14,976 KB
testcase_18 AC 375 ms
14,976 KB
testcase_19 AC 379 ms
14,848 KB
testcase_20 AC 378 ms
14,848 KB
testcase_21 AC 373 ms
14,848 KB
testcase_22 AC 373 ms
14,848 KB
testcase_23 AC 367 ms
14,848 KB
testcase_24 AC 377 ms
14,848 KB
testcase_25 AC 372 ms
14,976 KB
testcase_26 AC 366 ms
14,976 KB
testcase_27 AC 359 ms
14,208 KB
testcase_28 AC 361 ms
14,208 KB
testcase_29 AC 363 ms
14,080 KB
testcase_30 AC 364 ms
14,208 KB
testcase_31 AC 368 ms
14,208 KB
testcase_32 AC 323 ms
13,184 KB
testcase_33 AC 170 ms
8,192 KB
testcase_34 AC 163 ms
7,680 KB
testcase_35 AC 4 ms
5,248 KB
testcase_36 AC 339 ms
13,440 KB
testcase_37 AC 187 ms
8,576 KB
testcase_38 AC 169 ms
8,064 KB
testcase_39 AC 236 ms
10,368 KB
testcase_40 AC 134 ms
6,784 KB
testcase_41 AC 258 ms
10,752 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::collections::BTreeSet;

#[allow(unused_imports)]
use proconio::{
    input,
    marker::{Bytes, Chars, Usize1},
};

fn main() {
    input! {
        n: usize,
        mut p: [usize; n],
        a: [usize; n],
    }
    p.insert(0, 0);
    p.push(0);
    let mut q = vec![0; n + 1];
    for i in 1..=n {
        q[p[i]] = i;
    }
    let mut csum = vec![0; p.len() + 1];
    for i in 0..p.len() {
        csum[i + 1] = csum[i] + p[i];
    }
    let mut st = BTreeSet::from([0, n + 1]);
    for i in 1..=n {
        let m = q[i];
        let mut s = a[i - 1];
        let l = *st.range(..m).next_back().unwrap();
        let r = *st.range(m..).next().unwrap();
        st.insert(m);
        let mut res = 0;
        if m - l <= r - m {
            for &x in p[l + 1..=m].iter().rev() {
                if s < x {
                    break;
                }
                s -= x;
                res += csum[m + 1..=r].partition_point(|&c| c - csum[m + 1] <= s);
            }
        } else {
            for &x in &p[m..r] {
                if s < x {
                    break;
                }
                s -= x;
                let t = (m - l) - csum[l + 1..=m].partition_point(|&c| csum[m] - c > s);
                res += t;
            }
        }
        println!("{}", res);
    }
}
0