結果

問題 No.1341 真ん中を入れ替えて門松列
ユーザー akakimidori
提出日時 2021-01-15 23:28:25
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 1,303 ms / 2,000 ms
コード長 3,409 bytes
コンパイル時間 15,395 ms
コンパイル使用メモリ 378,072 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-26 18:32:50
合計ジャッジ時間 26,442 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

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

// ---------- begin input macro ----------
// reference: https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8
macro_rules! input {
(source = $s:expr, $($r:tt)*) => {
let mut iter = $s.split_whitespace();
input_inner!{iter, $($r)*}
};
($($r:tt)*) => {
let s = {
use std::io::Read;
let mut s = String::new();
std::io::stdin().read_to_string(&mut s).unwrap();
s
};
let mut iter = s.split_whitespace();
input_inner!{iter, $($r)*}
};
}
macro_rules! input_inner {
($iter:expr) => {};
($iter:expr, ) => {};
($iter:expr, $var:ident : $t:tt $($r:tt)*) => {
let $var = read_value!($iter, $t);
input_inner!{$iter $($r)*}
};
}
macro_rules! read_value {
($iter:expr, ( $($t:tt),* )) => {
( $(read_value!($iter, $t)),* )
};
($iter:expr, [ $t:tt ; $len:expr ]) => {
(0..$len).map(|_| read_value!($iter, $t)).collect::<Vec<_>>()
};
($iter:expr, chars) => {
read_value!($iter, String).chars().collect::<Vec<char>>()
};
($iter:expr, bytes) => {
read_value!($iter, String).bytes().collect::<Vec<u8>>()
};
($iter:expr, usize1) => {
read_value!($iter, usize) - 1
};
($iter:expr, $t:ty) => {
$iter.next().unwrap().parse::<$t>().expect("Parse error")
};
}
// ---------- end input macro ----------
fn run() {
input! {
n: usize,
m: i64,
p: [(i64, i64, i64); n],
}
let mut b = p.iter().map(|p| p.1).collect::<Vec<_>>();
b.sort();
let mut p = p.into_iter().map(|p| (p.0.min(p.2), p.0.max(p.2))).collect::<Vec<_>>();
p.sort();
let p = p;
let mut ans: Option<i64> = None;
for i in 0..=n {
let (l, r) = b.split_at(i);
let mut e = vec![];
for l in l.iter() {
e.push((*l, 1, 0));
}
for (k, &(a, _)) in p.iter().enumerate() {
e.push((a, 0, k));
}
e.sort();
let mut h = std::collections::BinaryHeap::<(i64, usize)>::new();
let mut cnt = 0;
let mut sum = 0;
for (_, op, k) in e.into_iter().rev() {
if op == 1 {
if let Some((c, _k)) = h.pop() {
sum += c;
cnt += 1;
}
} else {
h.push((p[k].1, k));
}
}
let p = h.into_iter().map(|(_, k)| p[k].1).collect::<Vec<_>>();
let mut e = vec![];
for r in r.iter() {
e.push((*r, 0));
}
for &c in p.iter() {
e.push((c, 1));
}
e.sort();
let mut stack = 0;
for (c, op) in e {
if op == 0 {
if stack > 0 {
cnt += 1;
sum += c;
stack -= 1;
}
} else {
stack += 1;
}
}
if cnt == n {
if ans.is_none() {
ans = Some(sum);
} else {
let v = ans.unwrap();
ans = Some(v.max(sum));
}
}
}
if let Some(ans) = ans {
println!("YES");
if ans >= m {
println!("KADOMATSU!");
} else {
println!("NO");
}
} else {
println!("NO");
}
}
fn main() {
run();
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0