結果
問題 | No.2727 Tetrahedron Game |
ユーザー | koba-e964 |
提出日時 | 2024-04-13 13:22:46 |
言語 | Rust (1.77.0 + proconio) |
結果 |
AC
|
実行時間 | 42 ms / 2,000 ms |
コード長 | 2,795 bytes |
コンパイル時間 | 13,392 ms |
コンパイル使用メモリ | 379,224 KB |
実行使用メモリ | 25,716 KB |
最終ジャッジ日時 | 2024-10-03 00:04:43 |
合計ジャッジ時間 | 13,592 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
5,248 KB |
testcase_01 | AC | 42 ms
5,248 KB |
testcase_02 | AC | 15 ms
5,248 KB |
testcase_03 | AC | 19 ms
25,664 KB |
testcase_04 | AC | 19 ms
25,716 KB |
testcase_05 | AC | 19 ms
12,892 KB |
testcase_06 | AC | 22 ms
14,604 KB |
testcase_07 | AC | 22 ms
8,860 KB |
testcase_08 | AC | 18 ms
11,016 KB |
testcase_09 | AC | 20 ms
10,496 KB |
ソースコード
use std::cmp::*; use std::io::Read; fn get_word() -> String { let stdin = std::io::stdin(); let mut stdin=stdin.lock(); let mut u8b: [u8; 1] = [0]; loop { let mut buf: Vec<u8> = Vec::with_capacity(16); loop { let res = stdin.read(&mut u8b); if res.unwrap_or(0) == 0 || u8b[0] <= b' ' { break; } else { buf.push(u8b[0]); } } if buf.len() >= 1 { let ret = String::from_utf8(buf).unwrap(); return ret; } } } fn get<T: std::str::FromStr>() -> T { get_word().parse().ok().unwrap() } fn mo(a: i64) -> i64 { (a % 606 + 606) % 606 } fn det3(mat: &[Vec<i64>]) -> i64 { let mut ret = 0; for i in 0..3 { let mut prod = 1; for j in 0..3 { prod *= mat[j][(i + j) % 3]; } ret += prod; } for i in 0..3 { let mut prod = 1; for j in 0..3 { prod *= mat[j][(i + 2 * j) % 3]; } ret -= prod; } ret } // https://yukicoder.me/problems/no/2727 (3.5) // 操作において j = k であれば体積が A[i]+1 倍になり、そうでなければ体積はそのままである。 // そのため実質的には、各プレイヤーが体積を A[i]+1 倍するかどうか、というゲームである。 // 6V mod 606 は 606 通りしかないので、これをキーにして DP をすればよい。 // Tags: dp fn main() { let t: i32 = get(); for _ in 0..t { let n: usize = get(); let k: usize = get(); let mut p = vec![vec![0i64; 3]; 3]; for i in 0..3 { for j in 0..3 { p[i][j] = get(); } } let a: Vec<usize> = (0..n).map(|_| get()).collect(); let s: Vec<_> = get_word().chars().collect(); let mut dp = vec![[0; 606]; n + 1]; for i in 0..k { dp[n][6 * i] = -1; } for i in k..101 { dp[n][6 * i] = 1; } for i in (0..n).rev() { if s[i] == 'K' { for j in 0..606 { let mut ma = dp[i + 1][j]; ma = max(ma, dp[i + 1][j * (a[i] + 1) % 606]); dp[i][j] = ma; } } else { for j in 0..606 { let mut mi = dp[i + 1][j]; mi = min(mi, dp[i + 1][j * (a[i] + 1) % 606]); dp[i][j] = mi; } } } let vraw = det3(&p).abs(); if vraw == 0 { println!("D"); continue; } let v = mo(vraw) as usize; println!("{}", if dp[0][v] == 1 { "K" } else if dp[0][v] == 0 { "D" } else { "P" }); } }