結果
問題 | No.2596 Christmas Eve (Heuristic ver.) |
ユーザー | EvbCFfp1XB |
提出日時 | 2023-12-24 23:01:59 |
言語 | Rust (1.77.0 + proconio) |
結果 |
TLE
|
実行時間 | - |
コード長 | 20,554 bytes |
コンパイル時間 | 2,665 ms |
コンパイル使用メモリ | 198,552 KB |
実行使用メモリ | 13,480 KB |
スコア | 0 |
最終ジャッジ日時 | 2023-12-24 23:05:30 |
合計ジャッジ時間 | 74,569 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | TLE | - |
testcase_01 | TLE | - |
testcase_02 | TLE | - |
testcase_03 | TLE | - |
testcase_04 | TLE | - |
testcase_05 | TLE | - |
testcase_06 | TLE | - |
testcase_07 | TLE | - |
testcase_08 | TLE | - |
testcase_09 | TLE | - |
testcase_10 | TLE | - |
testcase_11 | TLE | - |
testcase_12 | TLE | - |
testcase_13 | TLE | - |
testcase_14 | TLE | - |
testcase_15 | TLE | - |
testcase_16 | TLE | - |
testcase_17 | TLE | - |
testcase_18 | TLE | - |
testcase_19 | TLE | - |
testcase_20 | TLE | - |
testcase_21 | TLE | - |
testcase_22 | TLE | - |
testcase_23 | TLE | - |
testcase_24 | TLE | - |
testcase_25 | TLE | - |
testcase_26 | TLE | - |
testcase_27 | TLE | - |
testcase_28 | TLE | - |
testcase_29 | TLE | - |
testcase_30 | TLE | - |
testcase_31 | TLE | - |
testcase_32 | TLE | - |
testcase_33 | TLE | - |
testcase_34 | TLE | - |
testcase_35 | TLE | - |
testcase_36 | TLE | - |
testcase_37 | TLE | - |
testcase_38 | TLE | - |
testcase_39 | TLE | - |
testcase_40 | TLE | - |
testcase_41 | TLE | - |
testcase_42 | TLE | - |
testcase_43 | TLE | - |
testcase_44 | TLE | - |
testcase_45 | TLE | - |
testcase_46 | TLE | - |
testcase_47 | TLE | - |
testcase_48 | TLE | - |
testcase_49 | TLE | - |
testcase_50 | TLE | - |
testcase_51 | TLE | - |
testcase_52 | TLE | - |
testcase_53 | TLE | - |
testcase_54 | TLE | - |
testcase_55 | TLE | - |
testcase_56 | TLE | - |
testcase_57 | TLE | - |
testcase_58 | TLE | - |
testcase_59 | TLE | - |
testcase_60 | TLE | - |
testcase_61 | TLE | - |
testcase_62 | TLE | - |
testcase_63 | TLE | - |
testcase_64 | TLE | - |
testcase_65 | TLE | - |
testcase_66 | TLE | - |
testcase_67 | TLE | - |
testcase_68 | TLE | - |
testcase_69 | TLE | - |
testcase_70 | TLE | - |
testcase_71 | TLE | - |
testcase_72 | TLE | - |
testcase_73 | TLE | - |
testcase_74 | TLE | - |
testcase_75 | TLE | - |
testcase_76 | TLE | - |
testcase_77 | TLE | - |
testcase_78 | TLE | - |
testcase_79 | TLE | - |
testcase_80 | TLE | - |
testcase_81 | TLE | - |
testcase_82 | TLE | - |
testcase_83 | TLE | - |
testcase_84 | TLE | - |
testcase_85 | TLE | - |
testcase_86 | TLE | - |
testcase_87 | TLE | - |
testcase_88 | TLE | - |
testcase_89 | TLE | - |
testcase_90 | TLE | - |
testcase_91 | TLE | - |
testcase_92 | TLE | - |
testcase_93 | TLE | - |
testcase_94 | TLE | - |
testcase_95 | TLE | - |
testcase_96 | TLE | - |
testcase_97 | TLE | - |
testcase_98 | TLE | - |
testcase_99 | TLE | - |
testcase_100 | TLE | - |
testcase_101 | TLE | - |
testcase_102 | TLE | - |
testcase_103 | TLE | - |
testcase_104 | TLE | - |
testcase_105 | TLE | - |
testcase_106 | TLE | - |
testcase_107 | TLE | - |
testcase_108 | TLE | - |
testcase_109 | TLE | - |
testcase_110 | TLE | - |
testcase_111 | TLE | - |
testcase_112 | TLE | - |
testcase_113 | TLE | - |
testcase_114 | TLE | - |
testcase_115 | TLE | - |
testcase_116 | TLE | - |
testcase_117 | TLE | - |
testcase_118 | TLE | - |
testcase_119 | TLE | - |
testcase_120 | TLE | - |
testcase_121 | TLE | - |
testcase_122 | TLE | - |
testcase_123 | TLE | - |
testcase_124 | TLE | - |
ソースコード
use std::io; pub mod random { pub struct PcgXshRr { state: u64, } impl PcgXshRr { pub fn new(state: u64) -> Self { PcgXshRr { state } } pub fn next_int(&mut self) -> u32 { let oldstate = self.state; self.state = oldstate * 6364136223846793005u64 + 521u64; let xorshift: u32 = (((oldstate >> 18) ^ oldstate) >> 27) as u32; let rotation: u32 = (oldstate >> 59) as u32; return (xorshift >> rotation) | (xorshift << ((-1 * rotation as i32) & 31)); } pub fn next_double(&mut self) -> f64 { return (self.next_int() as f64) * 0.00000000023283064365386963; } pub fn gen(&mut self) -> usize { return self.next_int() as usize; } } } fn get_time() -> f64 { static mut START: f64 = -1.0; let time = std::time::SystemTime::now() .duration_since(std::time::UNIX_EPOCH) .unwrap() .as_secs_f64(); unsafe { if START < 0.0 { START = time; } time - START } } fn main() { get_time(); let mut input_line = String::new(); io::stdin().read_line(&mut input_line).unwrap(); let mut iter = input_line.split_whitespace(); let n: usize = iter.next().unwrap().parse().unwrap(); let k: usize = iter.next().unwrap().parse().unwrap(); let mut a = vec![0; n]; let mut b = vec![0; n]; let mut c = vec![0; 2 * n]; let mut d = vec![0; 2 * n]; let mut e = vec![0; n]; let mut f = vec![0; n]; input_line.clear(); io::stdin().read_line(&mut input_line).unwrap(); let mut iter = input_line.split_whitespace(); for i in 0..n { a[i] = iter.next().unwrap().parse().unwrap(); } input_line.clear(); io::stdin().read_line(&mut input_line).unwrap(); let mut iter = input_line.split_whitespace(); for i in 0..n { b[i] = iter.next().unwrap().parse().unwrap(); } input_line.clear(); io::stdin().read_line(&mut input_line).unwrap(); let mut iter = input_line.split_whitespace(); for i in 0..(2 * n) { c[i] = iter.next().unwrap().parse().unwrap(); } input_line.clear(); io::stdin().read_line(&mut input_line).unwrap(); let mut iter = input_line.split_whitespace(); for i in 0..(2 * n) { d[i] = iter.next().unwrap().parse().unwrap(); } input_line.clear(); io::stdin().read_line(&mut input_line).unwrap(); let mut iter = input_line.split_whitespace(); for i in 0..n { e[i] = iter.next().unwrap().parse().unwrap(); } input_line.clear(); io::stdin().read_line(&mut input_line).unwrap(); let mut iter = input_line.split_whitespace(); for i in 0..n { f[i] = iter.next().unwrap().parse().unwrap(); } let mut rng = random::PcgXshRr::new(5342); let mut u = (0..n).collect::<Vec<_>>(); let mut x = (0..n).collect::<Vec<_>>(); u.sort_by_key(|&i| a[i]); x.sort_by_key(|&i| e[i]); let mut vw = (0..2 * n).collect::<Vec<_>>(); vw.sort_by_key(|&i| c[i]); let mut v = (0..2 * n) .step_by(2) .map(|i| vw[i]) .skip(n - k) .chain((0..2 * n).step_by(2).map(|i| vw[i]).take(n - k)) .collect::<Vec<_>>(); let mut w = (1..2 * n) .step_by(2) .map(|i| vw[i]) .skip(n - k) .chain((1..2 * n).step_by(2).map(|i| vw[i]).take(n - k)) .collect::<Vec<_>>(); let mut heights = (0..n) .map(|i| a[u[i]] + c[v[i]] + c[w[i]] + e[x[i]]) .collect::<Vec<_>>(); for i in 0..k { for j in i + 1..k { if heights[j - 1] > heights[j] { u.swap(j - 1, j); v.swap(j - 1, j); w.swap(j - 1, j); x.swap(j - 1, j); heights.swap(j - 1, j); } } } { let start_temperature = 2e1; let end_temperature = 1e-9; let mut temperature = start_temperature; let max_iterations = 1e5 as usize; for iterations in 0..max_iterations { if (iterations & ((1 << 7) - 1)) == ((1 << 7) - 1) { temperature = end_temperature + (start_temperature - end_temperature) * (max_iterations - iterations) as f64 / (max_iterations) as f64; } let uvx = rng.gen() % 3; if uvx == 0 { let i = rng.gen() % k; let mut j = rng.gen() % (n - 1); if j >= i { j += 1; } if e[x[i]] < a[u[j]] && a[u[j]] < c[v[i]] { } else { continue; } if j < k { if e[x[j]] < a[u[i]] && a[u[i]] < c[v[j]] { } else { continue; } } let height_i = b[u[j]] + d[v[i]] + d[w[i]] + f[x[i]]; let height_j = b[u[i]] + d[v[j]] + d[w[j]] + f[x[j]]; let mut max = heights[k - 1].max(height_i); if j < k { max = max.max(height_j); } let mut min = heights[0].min(height_i); if j < k { min = min.min(height_j); } let delta_score = (40000 - (max - min)) - (40000 - (heights[k - 1] - heights[0])); if delta_score >= 0 || (delta_score as f64 / temperature >= -10.0 && rng.next_double() < (delta_score as f64 / temperature).exp()) { u.swap(i, j); heights[i] = height_i; heights[j] = height_j; sort(k, &mut u, &mut v, &mut w, &mut x, &mut heights); } } else if uvx == 1 { let i = rng.gen() % k; let mut j = rng.gen() % (n - 1); if j >= i { j += 1; } let vw1 = rng.gen() % 2; let vw2 = rng.gen() % 2; if vw1 == 0 { if vw2 == 0 { v.swap(i, j); } else { let swap = v[i]; v[i] = w[j]; w[j] = swap; } } else { if vw2 == 0 { let swap = w[i]; w[i] = v[j]; v[j] = swap; } else { w.swap(i, j); } } if a[u[i]] < c[v[i]].min(c[w[i]]) { } else { if vw1 == 0 { if vw2 == 0 { v.swap(i, j); } else { let swap = v[i]; v[i] = w[j]; w[j] = swap; } } else { if vw2 == 0 { let swap = w[i]; w[i] = v[j]; v[j] = swap; } else { w.swap(i, j); } } continue; } if j < k { if a[u[j]] < c[v[j]].min(c[w[j]]) { } else { if vw1 == 0 { if vw2 == 0 { v.swap(i, j); } else { let swap = v[i]; v[i] = w[j]; w[j] = swap; } } else { if vw2 == 0 { let swap = w[i]; w[i] = v[j]; v[j] = swap; } else { w.swap(i, j); } } continue; } } let height_i = b[u[i]] + d[v[i]] + d[w[i]] + f[x[i]]; let height_j = b[u[j]] + d[v[j]] + d[w[j]] + f[x[j]]; let mut max = heights[k - 1].max(height_i); if j < k { max = max.max(height_j); } let mut min = heights[0].min(height_i); if j < k { min = min.min(height_j); } let delta_score = (40000 - (max - min)) - (40000 - (heights[k - 1] - heights[0])); if delta_score >= 0 || (delta_score as f64 / temperature >= -10.0 && rng.next_double() < (delta_score as f64 / temperature).exp()) { if c[v[i]] > c[w[i]] { let swap = v[i]; v[i] = w[i]; w[i] = swap; } if c[v[j]] > c[w[j]] { let swap = v[j]; v[j] = w[j]; w[j] = swap; } heights[i] = height_i; heights[j] = height_j; sort(k, &mut u, &mut v, &mut w, &mut x, &mut heights); } else { if vw1 == 0 { if vw2 == 0 { v.swap(i, j); } else { let swap = v[i]; v[i] = w[j]; w[j] = swap; } } else { if vw2 == 0 { let swap = w[i]; w[i] = v[j]; v[j] = swap; } else { w.swap(i, j); } } } } else if uvx == 2 { let i = rng.gen() % k; let mut j = rng.gen() % (n - 1); if j >= i { j += 1; } if e[x[j]] < a[u[i]] { } else { continue; } if j < k { if e[x[i]] < a[u[j]] { } else { continue; } } let height_i = b[u[i]] + d[v[i]] + d[w[i]] + f[x[j]]; let height_j = b[u[j]] + d[v[j]] + d[w[j]] + f[x[i]]; let mut max = heights[k - 1].max(height_i); if j < k { max = max.max(height_j); } let mut min = heights[0].min(height_i); if j < k { min = min.min(height_j); } let delta_score = (40000 - (max - min)) - (40000 - (heights[k - 1] - heights[0])); if delta_score >= 0 || (delta_score as f64 / temperature >= -10.0 && rng.next_double() < (delta_score as f64 / temperature).exp()) { x.swap(i, j); heights[i] = height_i; heights[j] = height_j; sort(k, &mut u, &mut v, &mut w, &mut x, &mut heights); } } } } { let start_temperature = 2e1; let end_temperature = 1e-9; let mut temperature = start_temperature; let max_iterations = 2e7 as usize; for iterations in 0..max_iterations { if (iterations & ((1 << 7) - 1)) == ((1 << 7) - 1) { temperature = end_temperature + (start_temperature - end_temperature) * (max_iterations - iterations) as f64 / (max_iterations) as f64; } let uvx = rng.gen() % 3; if uvx == 0 { let i = (rng.gen() % 2 + (k - 1)) % k; let mut j = rng.gen() % (n - 1); if j >= i { j += 1; } if e[x[i]] < a[u[j]] && a[u[j]] < c[v[i]] { } else { continue; } if j < k { if e[x[j]] < a[u[i]] && a[u[i]] < c[v[j]] { } else { continue; } } let height_i = b[u[j]] + d[v[i]] + d[w[i]] + f[x[i]]; let height_j = b[u[i]] + d[v[j]] + d[w[j]] + f[x[j]]; let mut max = heights[k - 1].max(height_i); if j < k { max = max.max(height_j); } let mut min = heights[0].min(height_i); if j < k { min = min.min(height_j); } let delta_score = (40000 - (max - min)) - (40000 - (heights[k - 1] - heights[0])); if delta_score >= 0 || (delta_score as f64 / temperature >= -10.0 && rng.next_double() < (delta_score as f64 / temperature).exp()) { u.swap(i, j); heights[i] = height_i; heights[j] = height_j; sort(k, &mut u, &mut v, &mut w, &mut x, &mut heights); } } else if uvx == 1 { let i = (rng.gen() % 2 + (k - 1)) % k; let mut j = rng.gen() % (n - 1); if j >= i { j += 1; } let vw1 = rng.gen() % 2; let vw2 = rng.gen() % 2; if vw1 == 0 { if vw2 == 0 { v.swap(i, j); } else { let swap = v[i]; v[i] = w[j]; w[j] = swap; } } else { if vw2 == 0 { let swap = w[i]; w[i] = v[j]; v[j] = swap; } else { w.swap(i, j); } } if a[u[i]] < c[v[i]].min(c[w[i]]) { } else { if vw1 == 0 { if vw2 == 0 { v.swap(i, j); } else { let swap = v[i]; v[i] = w[j]; w[j] = swap; } } else { if vw2 == 0 { let swap = w[i]; w[i] = v[j]; v[j] = swap; } else { w.swap(i, j); } } continue; } if j < k { if a[u[j]] < c[v[j]].min(c[w[j]]) { } else { if vw1 == 0 { if vw2 == 0 { v.swap(i, j); } else { let swap = v[i]; v[i] = w[j]; w[j] = swap; } } else { if vw2 == 0 { let swap = w[i]; w[i] = v[j]; v[j] = swap; } else { w.swap(i, j); } } continue; } } let height_i = b[u[i]] + d[v[i]] + d[w[i]] + f[x[i]]; let height_j = b[u[j]] + d[v[j]] + d[w[j]] + f[x[j]]; let mut max = heights[k - 1].max(height_i); if j < k { max = max.max(height_j); } let mut min = heights[0].min(height_i); if j < k { min = min.min(height_j); } let delta_score = (40000 - (max - min)) - (40000 - (heights[k - 1] - heights[0])); if delta_score >= 0 || (delta_score as f64 / temperature >= -10.0 && rng.next_double() < (delta_score as f64 / temperature).exp()) { if c[v[i]] > c[w[i]] { let swap = v[i]; v[i] = w[i]; w[i] = swap; } if c[v[j]] > c[w[j]] { let swap = v[j]; v[j] = w[j]; w[j] = swap; } heights[i] = height_i; heights[j] = height_j; sort(k, &mut u, &mut v, &mut w, &mut x, &mut heights); } else { if vw1 == 0 { if vw2 == 0 { v.swap(i, j); } else { let swap = v[i]; v[i] = w[j]; w[j] = swap; } } else { if vw2 == 0 { let swap = w[i]; w[i] = v[j]; v[j] = swap; } else { w.swap(i, j); } } } } else if uvx == 2 { let i = (rng.gen() % 2 + (k - 1)) % k; let mut j = rng.gen() % (n - 1); if j >= i { j += 1; } if e[x[j]] < a[u[i]] { } else { continue; } if j < k { if e[x[i]] < a[u[j]] { } else { continue; } } let height_i = b[u[i]] + d[v[i]] + d[w[i]] + f[x[j]]; let height_j = b[u[j]] + d[v[j]] + d[w[j]] + f[x[i]]; let mut max = heights[k - 1].max(height_i); if j < k { max = max.max(height_j); } let mut min = heights[0].min(height_i); if j < k { min = min.min(height_j); } let delta_score = (40000 - (max - min)) - (40000 - (heights[k - 1] - heights[0])); if delta_score >= 0 || (delta_score as f64 / temperature >= -10.0 && rng.next_double() < (delta_score as f64 / temperature).exp()) { x.swap(i, j); heights[i] = height_i; heights[j] = height_j; sort(k, &mut u, &mut v, &mut w, &mut x, &mut heights); } } } } eprint!( "score : {}, time : {}", 40000 - (heights[k - 1] - heights[0]), get_time() ); for i in 0..k { println!("{} {} {} {}", u[i] + 1, v[i] + 1, w[i] + 1, x[i] + 1); } } fn sort( k: usize, u: &mut [usize], v: &mut [usize], w: &mut [usize], x: &mut [usize], heights: &mut [i32], ) { for _ in 0..2 { for b in 1..k { let a = b - 1; if heights[a] > heights[b] { u.swap(a, b); v.swap(a, b); w.swap(a, b); x.swap(a, b); heights.swap(a, b); } } } for _ in 0..2 { for b in (1..k).rev() { let a = b - 1; if heights[a] > heights[b] { u.swap(a, b); v.swap(a, b); w.swap(a, b); x.swap(a, b); heights.swap(a, b); } } } }