結果

問題 No.1306 Exactly 2 Digits
コンテスト
ユーザー akakimidori
提出日時 2020-12-03 00:40:44
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 2,885 bytes
コンパイル時間 15,025 ms
コンパイル使用メモリ 401,012 KB
実行使用メモリ 25,716 KB
平均クエリ数 1236.78
最終ジャッジ日時 2024-07-17 07:42:44
合計ジャッジ時間 42,022 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 34 WA * 89
権限があれば一括ダウンロードができます

ソースコード

diff # 

// (a, b), (c, d) (1 <= a, c < N, 0 <= b, d < N)
// 1.5 * N * (N - 1)
//
// クエリのi, j は入れ替えても同じ返答
// i = 1, j = 2..(N^2 - N) で質問すると何がわかる？
// iがわかるか？
//
// N=2
// 2 3 -> -1 0
// 3 2 -> 0 1
//
// N=3

fn read() -> Vec<i32> {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).unwrap();
    s.trim().split_whitespace().map(|s| s.parse().unwrap()).collect()
}

fn run() {
    let n = read()[0] as usize;
    let mut cnt = 0;
    let mut query = |x: usize, y: usize| -> (i32, i32) {
        assert!(x != y);
        assert!(x < n * n - n);
        assert!(y < n * n - n);
        assert!(cnt < n * (n - 1) / 2 * 3);
        cnt += 1;
        println!("? {} {}", x + 1, y + 1);
        let a = read();
        (a[0], a[1])
    };
    let m = n as i32;
    let hash = |x: i32, y: i32| -> (i32, i32) {
        assert!(m <= x && x < m * m);
        assert!(m <= y && y < m * m);
        let mut p = x / m - y / m;
        let mut q = x % m - y % m;
        if p > q {
            std::mem::swap(&mut p, &mut q);
        }
        (p, q)
    };
    let mut memo = vec![(0, 0); n * n - n];
    for i in 1..(n * n - n) {
        memo[i] = query(0, i);
    }
    let mut key = memo.clone();
    key.remove(0);
    key.sort();
    let mut ans = vec![0; n * n - n];
    let m = n as i32;
    for i in m..(m * m) {
        let mut a = vec![];
        for j in m..(m * m) {
            if i == j {
                continue;
            }
            a.push(hash(i, j));
        }
        a.sort();
        if a == key {
            ans[0] = i;
            break;
        }
    }
    let mut map = std::collections::BTreeMap::new();
    for i in 1..(n * n - n) {
        map.entry(memo[i]).or_insert(vec![]).push(i);
    }
    for i in 1..(n * n - n) {
        if ans[i] != 0 {
            continue;
        }
        let index = map.get(&memo[i]).unwrap();
        if index.len() == 1 {
            for v in m..(m * m) {
                if v == ans[0] {
                    continue;
                }
                if hash(ans[0], v) == memo[i] {
                    ans[i] = v;
                    break;
                }
            }
        } else {
            let mut cond = vec![];
            for v in m..(m * m) {
                if v == ans[0] {
                    continue;
                }
                if hash(ans[0], v) == memo[i] {
                    cond.push(v);
                }
            }
            if query(index[0], index[1]) == hash(cond[0], cond[1]) {
                ans[index[0]] = cond[0];
                ans[index[1]] = cond[1];
            } else {
                ans[index[0]] = cond[1];
                ans[index[1]] = cond[0];
            }
        }
    }
    print!("!");
    for a in ans {
        print!(" {}", a);
    }
    println!();
}

fn main() {
    run();
}
0