結果

問題 No.3128 Isosceles Triangle
コンテスト
ユーザー 👑 ArcAki
提出日時 2025-04-25 21:51:37
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 142 ms / 2,500 ms
コード長 1,854 bytes
コンパイル時間 17,382 ms
コンパイル使用メモリ 397,064 KB
実行使用メモリ 27,640 KB
最終ジャッジ日時 2025-04-25 21:52:11
合計ジャッジ時間 18,398 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#[allow(unused_imports)]
use std::{
    convert::{Infallible, TryFrom, TryInto as _}, fmt::{self, Debug, Display, Formatter,},
    fs::{File}, hash::{Hash, Hasher}, iter::{Product, Sum}, marker::PhantomData,
    ops::{Add, AddAssign, Sub, SubAssign, Div, DivAssign, Mul, MulAssign, Neg, RangeBounds},
    str::FromStr, sync::{atomic::{self, AtomicU32, AtomicU64}, Once},
    collections::{*, btree_set::Range}, mem::{swap},
    cmp::{self, Reverse, Ordering, Eq, PartialEq, PartialOrd},
    thread::LocalKey, f64::consts::PI, time::Instant, cell::RefCell,
    io::{self, stdin, Read, read_to_string, BufWriter, BufReader, stdout, Write},
};
#[allow(unused_imports)]
use proconio::{input, input_interactive, marker::{*}};
#[allow(unused_imports)]
//use rand::{thread_rng, Rng, seq::SliceRandom};
#[allow(unused_imports)]
//use ac_library::{*};

#[allow(dead_code)]
const INF: i64 = 1<<62;
#[allow(dead_code)]
const MOD: i64 = 998244353;
#[allow(dead_code)]
const D: [(usize, usize); 4] = [(1, 0), (0, 1), (!0, 0), (0, !0)];
#[allow(dead_code)]
const D2: [(usize, usize); 8] = [(1, 0), (1, 1), (0, 1), (!0, 1), (!0, 0), (!0, !0), (0, !0), (1, !0)];

use proconio::fastout;
#[fastout]
fn main() {
    input! {
        n: usize,
        a: [i64; n],
    }
    let mut x = a.clone();
    x.sort(); x.dedup();
    let k = x.len();
    let mut cnt = vec![0; k];
    let mut dic = HashMap::new();
    let mut set = BTreeSet::new();
    for (i, &v) in x.iter().enumerate(){
        dic.insert(v, i);
        set.insert(v);
    }
    for &v in &a{
        cnt[dic[&v]] += 1i64;
    }
    let mut ac = vec![0; k+1];
    for (i, &v) in cnt.iter().enumerate(){
        ac[i+1] = ac[i]+v;
    }
    let mut ans = 0;
    for (&v, &x) in cnt.iter().zip(&x){
        ans += (ac[dic[set.range(..2*x).next_back().unwrap()]+1]-v)*v*(v-1)/2;
    }
    println!("{}", ans);
}
0