Submission #13600249


Source Code Expand

#![allow(non_snake_case)]
use proconio::{input};

fn main() {
    input! {
        N: usize,
        P: [usize; N * N]
    }
    use std::collections::{VecDeque};

    let mut dist = vec![vec![-1; N]; N];
    let mut queue = VecDeque::new();
    for i in 0..N {
        for j in 0..N {
            if i == 0 || j == 0 || i == N - 1 || j == N - 1 {
                dist[i][j] = 1;
                queue.push_back((i as i32, j as i32));
            }
        }
    }
    
    while let Some((y, x)) = queue.pop_front() {
        for (dy, dx) in &[(1, 0), (0, 1), (-1, 0), (0, -1)] {
            let ny = y + dy;
            let nx = x + dx;
            if ny < 0 || nx < 0 || ny >= N as i32 || nx >= N as i32 { continue }
            if dist[ny as usize][nx as usize] == -1 {
                dist[ny as usize][nx as usize] = dist[y as usize][x as usize] + 1;
                queue.push_back((ny, nx));
            }
        }
    }

    let mut ans = 0;
    let mut grid = vec![vec![true; N]; N];
    for p in P {
        let y = ((p - 1) / N) as i32;
        let x = ((p - 1) % N) as i32;
        grid[y as usize][x as usize] = false;
        dist[y as usize][x as usize] -= 1;
        ans += dist[y as usize][x as usize] as i64;
        let mut queue = VecDeque::new();
        queue.push_back((y, x));
        while let Some((y, x)) = queue.pop_front() {
            for (dy, dx) in &[(1, 0), (0, 1), (-1, 0), (0, -1)] {
                let ny = y + dy;
                let nx = x + dx;
                if ny < 0 || nx < 0 || ny >= N as i32 || nx >= N as i32 { continue }
                let cost = if grid[ny as usize][nx as usize] { 1 } else { 0 };
                if dist[ny as usize][nx as usize] > dist[y as usize][x as usize] + cost {
                    dist[ny as usize][nx as usize] = dist[y as usize][x as usize] + cost;
                    queue.push_back((ny, nx))
                }
            }
        }
    }
    println!("{}", ans);
}

Submission Info

Submission Time
Task B - Joker
User pekempey
Language Rust (1.42.0)
Score 700
Code Size 2009 Byte
Status AC
Exec Time 582 ms
Memory 6852 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 700 / 700
Status
AC × 3
AC × 24
Set Name Test Cases
Sample example0.txt, example1.txt, example2.txt
All 000.txt, 001.txt, 002.txt, 003.txt, 004.txt, 005.txt, 006.txt, 007.txt, 008.txt, 009.txt, 010.txt, 011.txt, 012.txt, 013.txt, 014.txt, 015.txt, 016.txt, 017.txt, 018.txt, 019.txt, 020.txt, example0.txt, example1.txt, example2.txt
Case Name Status Exec Time Memory
000.txt AC 11 ms 2344 KiB
001.txt AC 35 ms 2584 KiB
002.txt AC 410 ms 6792 KiB
003.txt AC 2 ms 2116 KiB
004.txt AC 2 ms 2136 KiB
005.txt AC 2 ms 1944 KiB
006.txt AC 2 ms 2056 KiB
007.txt AC 2 ms 2068 KiB
008.txt AC 2 ms 2056 KiB
009.txt AC 2 ms 2168 KiB
010.txt AC 2 ms 2112 KiB
011.txt AC 2 ms 2052 KiB
012.txt AC 2 ms 1924 KiB
013.txt AC 16 ms 2372 KiB
014.txt AC 19 ms 2308 KiB
015.txt AC 75 ms 3012 KiB
016.txt AC 571 ms 6568 KiB
017.txt AC 570 ms 6624 KiB
018.txt AC 574 ms 6644 KiB
019.txt AC 582 ms 6600 KiB
020.txt AC 582 ms 6852 KiB
example0.txt AC 2 ms 2148 KiB
example1.txt AC 2 ms 2084 KiB
example2.txt AC 2 ms 1996 KiB