#892315 (Rust) No.5017 Tool-assisted Shooting

提出ソース

結果

問題	No.5017 Tool-assisted Shooting
ユーザー	ntk-ta01
提出日時	2023-07-16 17:56:07
言語	Rust (1.77.0)
結果	RE
実行時間	-
コード長	5,138 bytes
コンパイル時間	1,476 ms
コンパイル使用メモリ	163,496 KB
実行使用メモリ	24,384 KB
スコア	74,830
平均クエリ数	128.85
最終ジャッジ日時	2023-07-16 17:56:33
合計ジャッジ時間	9,190 ms
ジャッジサーバーID （参考情報）	judge11 / judge14

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	31 ms 23,928 KB
testcase_01	AC	31 ms 24,300 KB
testcase_02	AC	27 ms 23,376 KB
testcase_03	AC	29 ms 23,544 KB
testcase_04	AC	29 ms 23,388 KB
testcase_05	AC	29 ms 24,276 KB
testcase_06	AC	28 ms 24,060 KB
testcase_07	AC	30 ms 23,436 KB
testcase_08	AC	27 ms 23,652 KB
testcase_09	RE	-
testcase_10	AC	30 ms 23,544 KB
testcase_11	AC	29 ms 23,952 KB
testcase_12	AC	26 ms 24,048 KB
testcase_13	AC	27 ms 23,436 KB
testcase_14	AC	27 ms 24,264 KB
testcase_15	AC	27 ms 23,568 KB
testcase_16	AC	26 ms 23,568 KB
testcase_17	AC	30 ms 23,568 KB
testcase_18	AC	30 ms 23,568 KB
testcase_19	AC	31 ms 24,060 KB
testcase_20	AC	29 ms 23,568 KB
testcase_21	AC	45 ms 23,652 KB
testcase_22	AC	29 ms 23,616 KB
testcase_23	AC	27 ms 24,240 KB
testcase_24	AC	37 ms 24,048 KB
testcase_25	AC	28 ms 24,300 KB
testcase_26	RE	-
testcase_27	AC	30 ms 23,544 KB
testcase_28	RE	-
testcase_29	AC	27 ms 23,736 KB
testcase_30	AC	30 ms 23,940 KB
testcase_31	AC	26 ms 23,376 KB
testcase_32	AC	31 ms 23,652 KB
testcase_33	AC	28 ms 24,240 KB
testcase_34	AC	30 ms 23,400 KB
testcase_35	AC	27 ms 23,556 KB
testcase_36	AC	36 ms 24,024 KB
testcase_37	AC	28 ms 23,568 KB
testcase_38	RE	-
testcase_39	AC	31 ms 23,532 KB
testcase_40	AC	29 ms 23,748 KB
testcase_41	AC	27 ms 23,412 KB
testcase_42	AC	39 ms 23,388 KB
testcase_43	AC	31 ms 23,292 KB
testcase_44	AC	28 ms 23,964 KB
testcase_45	RE	-
testcase_46	AC	30 ms 24,072 KB
testcase_47	AC	32 ms 24,168 KB
testcase_48	AC	31 ms 23,304 KB
testcase_49	AC	33 ms 23,832 KB
testcase_50	RE	-
testcase_51	AC	30 ms 23,628 KB
testcase_52	AC	34 ms 24,264 KB
testcase_53	AC	31 ms 23,748 KB
testcase_54	RE	-
testcase_55	AC	44 ms 23,844 KB
testcase_56	AC	27 ms 23,736 KB
testcase_57	AC	32 ms 24,060 KB
testcase_58	AC	31 ms 23,748 KB
testcase_59	AC	27 ms 23,952 KB
testcase_60	AC	36 ms 24,240 KB
testcase_61	AC	31 ms 24,300 KB
testcase_62	AC	29 ms 23,664 KB
testcase_63	AC	28 ms 24,036 KB
testcase_64	AC	29 ms 23,436 KB
testcase_65	AC	31 ms 23,928 KB
testcase_66	AC	32 ms 23,952 KB
testcase_67	AC	28 ms 24,024 KB
testcase_68	AC	29 ms 24,252 KB
testcase_69	AC	30 ms 23,544 KB
testcase_70	AC	28 ms 23,580 KB
testcase_71	AC	28 ms 23,436 KB
testcase_72	AC	28 ms 23,424 KB
testcase_73	AC	30 ms 23,292 KB
testcase_74	AC	32 ms 24,048 KB
testcase_75	AC	28 ms 23,952 KB
testcase_76	AC	27 ms 23,388 KB
testcase_77	AC	28 ms 23,916 KB
testcase_78	AC	31 ms 23,940 KB
testcase_79	AC	33 ms 24,264 KB
testcase_80	AC	32 ms 23,292 KB
testcase_81	AC	29 ms 24,384 KB
testcase_82	AC	30 ms 24,024 KB
testcase_83	AC	29 ms 23,664 KB
testcase_84	AC	27 ms 24,312 KB
testcase_85	AC	28 ms 23,556 KB
testcase_86	AC	30 ms 24,264 KB
testcase_87	AC	26 ms 23,940 KB
testcase_88	AC	28 ms 24,252 KB
testcase_89	AC	31 ms 24,312 KB
testcase_90	AC	28 ms 23,388 KB
testcase_91	AC	27 ms 23,328 KB
testcase_92	AC	30 ms 24,180 KB
testcase_93	RE	-
testcase_94	AC	31 ms 23,568 KB
testcase_95	RE	-
testcase_96	AC	30 ms 24,300 KB
testcase_97	AC	26 ms 23,400 KB
testcase_98	AC	27 ms 23,844 KB
testcase_99	AC	30 ms 24,264 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

use std::{
    collections::HashMap,
    io::{stdin, stdout, Write},
};

const H: usize = 60;
const W: usize = 25;

// type Output = Vec<char>;

fn main() {
    let mut sim = Sim::new();
    for _ in 0..1000 {
        if let Some(input) = read_input() {
            let is_defeat = sim.proceed(input.es, 'S');
            println!("S");
            stdout().flush().unwrap();
            if is_defeat {
                break;
            }
        } else {
            break;
        }
    }
}

fn read_input() -> Option<Input> {
    let mut n = String::new();
    stdin().read_line(&mut n).unwrap();
    if n == "-1" {
        return None;
    }
    let n = n.trim().parse::<usize>().unwrap();
    let mut es = vec![];
    for _ in 0..n {
        let mut line = String::new();
        stdin().read_line(&mut line).unwrap();
        let mut tokens = line.split_whitespace();
        let h = tokens.next().unwrap().parse::<i64>().unwrap();
        let p = tokens.next().unwrap().parse::<i64>().unwrap();
        let x = tokens.next().unwrap().parse::<usize>().unwrap();
        es.push((h, p, x));
    }

    Some(Input { n, es })
}

#[allow(dead_code)]
#[derive(Debug)]
struct Input {
    n: usize,
    es: Vec<(i64, i64, usize)>,
}

#[derive(Debug, Clone, Copy)]
pub struct Player {
    s: i64,
    pos_x: usize,
}

impl Player {
    pub fn new() -> Self {
        Player { s: 0, pos_x: 12 }
    }

    pub fn level(&self) -> i64 {
        1 + self.s / 100
    }
}

impl Default for Player {
    fn default() -> Self {
        Self::new()
    }
}

struct Enemy {
    hp: i64,
    dameged: i64,
    power: i64,
}

impl Enemy {
    pub fn new(hp: i64, power: i64) -> Self {
        Self {
            hp,
            dameged: 0,
            power,
        }
    }
}

pub struct Sim {
    board: Vec<Vec<Square>>,
    player: Player,
    enemies: HashMap<usize, Enemy>,
    score: i64,
}

impl Sim {
    pub fn new() -> Self {
        let mut board = vec![vec![Square::EMPTY; W]; H];
        board[0][12] = Square::PLAYER;
        let player = Player::new();
        let enemies = HashMap::new();
        Self {
            board,
            player,
            enemies,
            score: 0,
        }
    }

    pub fn proceed(&mut self, es: Vec<(i64, i64, usize)>, command: char) -> bool {
        let mut is_defeat = false;
        // 全enemyが一段下に
        for h in 0..H {
            for w in 0..W {
                if let Square::ENEMY { id } = self.board[h][w] {
                    if h == 0 {
                        self.board[h][w] = Square::EMPTY;
                    } else {
                        self.board[h][w] = Square::EMPTY;
                        if Square::PLAYER == self.board[h - 1][w] {
                            is_defeat = true;
                        }
                        self.board[h - 1][w] = Square::ENEMY { id };
                    }
                }
            }
        }
        // 新たなenemyが出現
        for (h, p, x) in es {
            let id = self.enemies.len() + 1;
            self.board[H - 1][x] = Square::ENEMY { id };
            self.enemies.insert(id, Enemy::new(h, p));
        }
        // playerが移動
        match command {
            'L' => {
                let prev_x = self.player.pos_x;
                let next_x = (self.player.pos_x + W - 1) % W;
                if let Square::ENEMY { id: _ } = self.board[0][next_x] {
                    is_defeat = true;
                } else {
                    self.board[0][next_x] = self.board[0][prev_x];
                }
                self.board[0][prev_x] = Square::EMPTY;
                self.player.pos_x = next_x;
            }
            'R' => {
                let prev_x = self.player.pos_x;
                let next_x = (self.player.pos_x + W + 1) % W;
                self.player.pos_x = next_x;
                if let Square::ENEMY { id: _ } = self.board[0][next_x] {
                    is_defeat = true;
                } else {
                    self.board[0][next_x] = self.board[0][prev_x];
                }
                self.board[0][prev_x] = Square::EMPTY;
                self.player.pos_x = next_x;
            }
            'S' => {}
            _ => unreachable!(),
        }
        // playerによる攻撃
        if !is_defeat {
            for h in 1..H {
                if let Square::ENEMY { id } = self.board[h][self.player.pos_x] {
                    let enemy = self.enemies.get_mut(&id).unwrap();
                    enemy.dameged += self.player.level();
                    if enemy.hp <= enemy.dameged {
                        self.player.s += enemy.power;
                        self.score += enemy.hp;
                        self.enemies.remove(&id);
                        self.board[h][self.player.pos_x] = Square::EMPTY;
                    }
                    break;
                }
            }
        }
        is_defeat
    }
}

impl Default for Sim {
    fn default() -> Self {
        Self::new()
    }
}

#[derive(Debug, Clone, Copy, PartialEq, PartialOrd)]
pub enum Square {
    PLAYER,
    ENEMY { id: usize },
    EMPTY,
}

yukicoder

結果

テストケース

ソースコード