結果

問題 No.911 ラッキーソート
ユーザー akakimidori
提出日時 2019-10-19 18:03:57
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 57 ms / 2,000 ms
コード長 1,995 bytes
コンパイル時間 12,933 ms
コンパイル使用メモリ 379,436 KB
実行使用メモリ 7,424 KB
最終ジャッジ日時 2024-06-27 14:23:37
合計ジャッジ時間 16,255 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::io::Read;

fn fill(k: usize, a: &[usize], cond: &mut [usize]) {
    if a.len() <= 1 {
        return;
    }
    let mut split = 0;
    for i in 1..a.len() {
        let p = (a[i - 1] >> k) & 1;
        let q = (a[i] >> k) & 1;
        if p == 0 && q == 1 {
            cond[k] &= 1;
            split = i;
        }
        if p == 1 && q == 0 {
            cond[k] &= 2;
            split = i;
        }
    }
    if k == 0 {
        if a.len() > 2 || split != 1 {
            cond[k] = 0;
        }
        return;
    }
    if cond[k] == 3 {
        fill(k - 1, a, cond);
    } else if cond[k] != 0 {
        fill(k - 1, &a[..split], cond);
        fill(k - 1, &a[split..], cond);
    }
}

fn run() {
    let mut s = String::new();
    std::io::stdin().read_to_string(&mut s).unwrap();
    let mut it = s.trim().split_whitespace();
    let n: usize = it.next().unwrap().parse().unwrap();
    let l: usize = it.next().unwrap().parse().unwrap();
    let r: usize = it.next().unwrap().parse().unwrap();
    let a: Vec<usize> = (0..n).map(|_| it.next().unwrap().parse().unwrap()).collect();
    let bit = 61;
    let mut cond = vec![3; bit];
    fill(bit - 1, &a, &mut cond);
    if cond.contains(&0) {
        println!("0");
        return;
    }
    let mut dp = [[0usize; 2]; 2];
    dp[0][0] = 1;
    for k in (0..bit).rev() {
        let mut next = [[0; 2]; 2];
        let l = (l >> k) & 1;
        let r = (r >> k) & 1;
        for i in 0..2 {
            for j in 0..2 {
                //0
                if cond[k] & 1 != 0 && (i > 0 || l == 0) {
                    next[i][j | r] += dp[i][j];
                }
                //1
                if cond[k] & 2 != 0 && (j > 0 || r == 1) {
                    next[i | (l ^ 1)][j] += dp[i][j];
                }
            }
        }
        dp = next;
    }
    let mut ans = 0;
    for i in 0..2 {
        for j in 0..2 {
            ans += dp[i][j];
        }
    }
    println!("{}", ans);
}

fn main() {
    run();
}
0