結果

問題 No.1282 Display Elements
ユーザー Strorkis
提出日時 2020-11-06 22:15:05
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 53 ms / 2,000 ms
コード長 2,253 bytes
コンパイル時間 13,092 ms
コンパイル使用メモリ 377,608 KB
実行使用メモリ 6,608 KB
最終ジャッジ日時 2024-07-22 13:01:01
合計ジャッジ時間 14,902 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

pub mod procon_input {
use std::{str::FromStr, iter::FromIterator};
pub fn read_line() -> String {
let mut input = String::new();
std::io::stdin().read_line(&mut input).ok();
input
}
pub fn parse<T: FromStr>(s: &str) -> T {
s.parse().ok().unwrap()
}
pub fn read<T: FromStr>() -> T {
parse(read_line().trim_end())
}
pub fn read_collection<T: FromStr, C: FromIterator<T>>() -> C {
read_line().split_whitespace().map(parse).collect()
}
#[macro_export]
macro_rules! read_tuple {
( $( $t:ty ),* ) => {{
let input = read_line();
let mut iter = input.split_whitespace();
( $( parse::<$t>(iter.next().unwrap()) ),* )
}};
}
}
struct FenwickTree<T> {
n: usize,
init: T,
data: Vec<T>,
}
impl<T: Copy + std::ops::AddAssign> FenwickTree<T> {
fn new(n: usize, init: T) -> FenwickTree<T> {
FenwickTree { n, init, data: vec![init; n] }
}
fn add(&mut self, mut p: usize, x: T) {
p += 1;
while p <= self.n {
self.data[p - 1] += x;
p += p & (!p + 1);
}
}
fn sum(&self, mut r: usize) -> T {
let mut res = self.init;
while r > 0 {
res += self.data[r - 1];
r -= r & (!r + 1);
}
res
}
}
use procon_input::*;
fn solve(writer: &mut std::io::BufWriter<std::io::StdoutLock>) {
use std::io::Write;
let n: usize = read();
let mut a: Vec<i32> = read_collection();
a.sort();
let b: Vec<i32> = read_collection();
let mut v: Vec<i32> = a.iter().chain(b.iter()).cloned().collect();
v.sort();
v.dedup();
let a: Vec<usize> = {
a.into_iter().map(|x| v.binary_search(&x).unwrap()).collect()
};
let b: Vec<usize> = {
b.into_iter().map(|x| v.binary_search(&x).unwrap()).collect()
};
let mut ans: i64 = 0;
let mut ft = FenwickTree::new(2 * n, 0);
for i in 0..n {
ft.add(b[i], 1);
ans += ft.sum(a[i]);
}
writeln!(writer, "{}", ans).ok();
}
fn main() {
let stdout = std::io::stdout();
let mut writer = std::io::BufWriter::new(stdout.lock());
solve(&mut writer);
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0