結果
問題 | No.270 next_permutation (1) |
ユーザー | akakimidori |
提出日時 | 2023-09-10 21:48:40 |
言語 | Rust (1.77.0) |
結果 |
AC
|
実行時間 | 14 ms / 2,000 ms |
コード長 | 2,206 bytes |
コンパイル時間 | 1,714 ms |
コンパイル使用メモリ | 148,496 KB |
実行使用メモリ | 5,504 KB |
最終ジャッジ日時 | 2023-09-10 21:48:44 |
合計ジャッジ時間 | 2,922 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,380 KB |
testcase_01 | AC | 2 ms
4,376 KB |
testcase_02 | AC | 2 ms
4,380 KB |
testcase_03 | AC | 1 ms
4,380 KB |
testcase_04 | AC | 2 ms
4,376 KB |
testcase_05 | AC | 2 ms
4,388 KB |
testcase_06 | AC | 2 ms
4,376 KB |
testcase_07 | AC | 1 ms
4,376 KB |
testcase_08 | AC | 1 ms
4,380 KB |
testcase_09 | AC | 3 ms
4,376 KB |
testcase_10 | AC | 12 ms
5,496 KB |
testcase_11 | AC | 13 ms
5,500 KB |
testcase_12 | AC | 14 ms
5,484 KB |
testcase_13 | AC | 12 ms
5,412 KB |
testcase_14 | AC | 12 ms
5,504 KB |
ソースコード
fn main() { input! { n: usize, k: usize, a: [i64; n], b: [i64; n], } let dist = |a: &[i64], b: &[i64]| -> i64 { a.iter().zip(b.iter()).fold(0, |s, p| s + (*p.0 - *p.1).abs()) }; let mut a = a; let mut ans = 0; let mut sum = dist(&a, &b); for _ in 0..k { ans += sum; if let Some(x) = a.windows(2).rposition(|a| a[0] < a[1]) { sum -= dist(&a[x..], &b[x..]); let y = a.iter().rposition(|p| *p > a[x]).unwrap(); a.swap(x, y); a[(x + 1)..].reverse(); sum += dist(&a[x..], &b[x..]); } else { a.reverse(); sum = dist(&a, &b); } } println!("{}", ans); } // ---------- begin input macro ---------- // reference: https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8 #[macro_export] 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_export] 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_export] 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 ----------