結果

問題 No.2453 Seat Allocation
ユーザー ikomaikoma
提出日時 2023-09-01 23:41:07
言語 Rust
(1.77.0)
結果
WA  
実行時間 -
コード長 1,640 bytes
コンパイル時間 1,147 ms
コンパイル使用メモリ 152,668 KB
実行使用メモリ 9,288 KB
最終ジャッジ日時 2023-09-01 23:41:13
合計ジャッジ時間 5,623 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
9,264 KB
testcase_01 AC 1 ms
4,380 KB
testcase_02 AC 1 ms
4,384 KB
testcase_03 AC 1 ms
4,380 KB
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 TLE -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::collections::BinaryHeap;
use std::cmp::Reverse;

fn main() {
    let mut input = String::new();
    std::io::stdin().read_line(&mut input).unwrap();
    let mut input_iter = input.split_whitespace();
    
    let _: usize = input_iter.next().unwrap().parse().unwrap();
    let _: usize = input_iter.next().unwrap().parse().unwrap();
    
    input.clear();
    std::io::stdin().read_line(&mut input).unwrap();
    let a: Vec<i64> = input.split_whitespace().map(|x| x.parse().unwrap()).collect();
    
    input.clear();
    std::io::stdin().read_line(&mut input).unwrap();
    let b: Vec<i64> = input.split_whitespace().map(|x| x.parse().unwrap()).collect();
    
    let mut ai: Vec<(i64, usize)> = a.iter().enumerate().map(|(i, &x)| (x, i)).collect();
    ai.sort_by_key(|x| Reverse(x.0));
    
    let mut heap: BinaryHeap<(i64, usize)> = BinaryHeap::new(); // u64型を使用
    
    let (a_max, i_max) = ai[0];
    for &b_j in b.iter() {
        let p = (a_max as f64 / b_j as f64 * 1e6) as i64; 
        heap.push((-p, i_max));
    }
    
    for &(a_i, i) in ai.iter().skip(1) {
        for &b_j in b.iter() {
            let p = (a_i as f64 / b_j as f64 * 1e6) as i64; // 小数点以下を考慮
            heap.push((-p, i)); // u64型を使用
            if let Some((p2, i2)) = heap.pop() {
                if p != -p2 || i != i2 {
                    break;
                }
            }
        }
    }
    
    let mut ans: Vec<usize> = Vec::new();
    while let Some((_, i)) = heap.pop() {
        ans.push(i+1);
    }
    
    ans.reverse();
    
    for &x in ans.iter() {
        println!("{}", x);
    }
}
0