結果

問題 No.2632 Center of Three Points in Lp Norm
コンテスト
ユーザー akakimidori
提出日時 2024-02-16 23:33:23
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 3,486 bytes
コンパイル時間 16,305 ms
コンパイル使用メモリ 401,232 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-09-28 22:29:16
合計ジャッジ時間 19,252 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 RE * 1
other AC * 23 WA * 11
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: function `ternary_search` is never used
  --> src/main.rs:61:4
   |
61 | fn ternary_search<F>(mut l: f64, mut r: f64, iter: usize, f: F) -> f64
   |    ^^^^^^^^^^^^^^
   |
   = note: `#[warn(dead_code)]` on by default

ソースコード

diff # 

fn main() {
    input! {
        p: f64,
        a: [(f64, f64); 3],
    }
    let iter = 500;
    let inf = 10f64.powi(10);
    let norm = |a: (f64, f64), b: (f64, f64)| -> f64 {
        ((a.0 - b.0).abs().powf(p) + (a.1 - b.1).abs().powf(p)).powf(1.0 / p)
    };
    // y, diff
    let f = |x: f64, p: usize, q: usize| -> (f64, f64) {
        let mut l = -inf;
        let mut r = inf;
        if norm(a[p], (x, l)) > norm(a[q], (x, l)) {
            std::mem::swap(&mut l, &mut r);
        }
        for _ in 0..iter {
            let m = (l + r) * 0.5;
            if norm(a[p], (x, m)) <= norm(a[q], (x, m)) {
                l = m;
            } else {
                r = m;
            }
        }
        let y = (l + r) * 0.5;
        let d = norm(a[p], (x, y)) - norm(a[3 ^ p ^ q], (x, y));
        (y, d)
    };
    let mut p = (3, 3);
    for i in 0..3 {
        for j in 0..3 {
            if i != j && a[i].0 != a[j].0 && f(-inf, i, j).1 > 0.0 && f(inf, i, j).1 < 0.0 {
                p = (i, j);
            }
        }
    }
    let (p, q) = p;
    let mut ans = (0.0, 0.0, std::f64::MAX);
    let mid = (a[p].0 + a[q].0) * 0.5;
    for &(mut l, mut r) in [(-inf, mid), (mid, inf)].iter() {
        let sign = f(l, p, q).1.signum();
        for _ in 0..iter {
            let m = (l + r) * 0.5;
            if f(m, p, q).1.signum() == sign {
                l = m;
            } else {
                r = m;
            }
        }
        let x = (l + r) * 0.5;
        let y = f(x, p, q).0;
        let key = norm((x, y), a[3 ^ p ^ q]);
        if key < ans.2 {
            ans = (x, y, key);
        }
    }
    println!("{:.9} {:.9}", ans.0, ans.1);
}

fn ternary_search<F>(mut l: f64, mut r: f64, iter: usize, f: F) -> f64
where
    F: Fn(f64) -> f64,
{
    let inv = 1.0f64 / 3.0;
    for _ in 0..iter {
        let ll = (l + l + r) * inv;
        let rr = (l + r + r) * inv;
        if f(ll) <= f(rr) {
            r = rr;
        } else {
            l = ll;
        }
    }
    (l + r) * 0.5
}

// ---------- 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 ----------
0