結果

問題 No.2596 Christmas Eve (Heuristic ver.)
ユーザー EvbCFfp1XBEvbCFfp1XB
提出日時 2023-12-24 23:01:59
言語 Rust
(1.77.0)
結果
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 -
権限があれば一括ダウンロードができます

ソースコード

diff #

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);
            }
        }
    }
}
0