結果

問題 No.3685 ワロングアンサーやんけ!
コンテスト
ユーザー ikoma
提出日時 2026-09-05 16:32:28
言語 Rust
(1.97.1 + proconio + num + itertools)
コンパイル:
/usr/bin/rustc_custom
実行:
./target/release/main
結果
AC  
実行時間 187 ms / 2,000 ms
+ 460µs
コード長 2,945 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,867 ms
コンパイル使用メモリ 196,944 KB
実行使用メモリ 6,272 KB
最終ジャッジ日時 2026-09-05 16:32:35
合計ジャッジ時間 5,523 ms
ジャッジサーバーID
(参考情報)
judge5_0 / judge1_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

#![allow(unused_imports, dead_code, unused_macros, unused_variables, non_snake_case, unused_parens)]
use proconio::{input,marker::{Bytes, Chars, Usize1,Isize1}};
use std::ops::*;
use std::cmp::*;
use std::mem::swap;
use std::collections::*;

const MOD:u64 = 1_000_000_007;
const INF:i64 = 0x3fff_ffff_ffff_ffff;
const YES:&str="Yes";const NO:&str="No";macro_rules!yesno{($var:expr) => {if $var{println!("{}",YES);}else{println!("{}",NO);}}}

macro_rules! min {($a:expr $(,)*) => {{$a}};($a:expr, $b:expr $(,)*) => {{std::cmp::min($a, $b)}};($a:expr, $($rest:expr),+ $(,)*) => {{std::cmp::min($a, min!($($rest),+))}};}
macro_rules! max {($a:expr $(,)*) => {{$a}};($a:expr, $b:expr $(,)*) => {{std::cmp::max($a, $b)}};($a:expr, $($rest:expr),+ $(,)*) => {{std::cmp::max($a, max!($($rest),+))}};}
macro_rules! chmin {($base:expr, $($cmps:expr),+ $(,)*) => {{let cmp_min = min!($($cmps),+);if $base > cmp_min {$base = cmp_min;true} else {false}}};}
macro_rules! chmax {($base:expr, $($cmps:expr),+ $(,)*) => {{let cmp_max = max!($($cmps),+);if $base < cmp_max {$base = cmp_max;true} else {false}}};}
macro_rules! mulvec {($x:expr; $s:expr) => {vec![$x; $s]};($x:expr; $s0:expr; $( $s:expr );+) => {mulvec![vec![$x; $s0]; $( $s );+ ]};}
macro_rules! outputln {($var:expr)=>{println!("{}",$var)};($var:expr,$($vars:expr),+)=>{print!("{} ",$var);outputln!($($vars),+);};}
macro_rules! debug {($($a:expr),* $(,)*) => {eprintln!(concat!($("| ",stringify!($a), "={:?} "),*, "|"),$(&$a),*);};}


fn solve() {
    input! {
        T: usize,
    }
	for _ in 0..T {
		input!{
			mut R: Chars,
			S: Chars,
			K: usize,
		}
		let topK_W = R.iter().take(K).filter(|r| **r == 'W').count();
		let topK_A = R.iter().take(K).filter(|r| **r == 'A').count();
		let topK_Q = R.iter().take(K).filter(|r| **r == '?').count();
		let last_W = R.iter().skip(K).filter(|r| **r == 'W').count();
		let last_A = R.iter().skip(K).filter(|r| **r == 'A').count();
		let last_Q = R.iter().skip(K).filter(|r| **r == '?').count();
		let include_W = (topK_W + last_W) > 0;
		if S[0] == 'N' {
			// 先頭Kが全てAではない or Wが1文字も含まれない
			let topK_is_A = (0..K).all(|i| R[i] != 'W');
			if include_W {
				// 先頭KはAのみではない
				if topK_W == 0 && topK_Q == 1 {
					for i in 0..K {
						if R[i] == '?' {
							R[i] = 'W';
						}
					}
				}
			}
			if topK_A == K {
				// すべてA確定
				for i in K..R.len() {
					R[i] = 'A';
				}
			}

		}else{
			// 先頭Kは'A'
			for i in 0..K {
				R[i] = 'A';
			}
			// Wが含まれる
			if !include_W {
				let mut cnt = 0;
				let mut idx = 0;
				for i in K..R.len() {
					if R[i] == '?' {
						cnt += 1;
						idx = i;
					}
				}
				if cnt == 1 {
					R[idx] = 'W';
				}
			}
		}
		outputln!(R.iter().collect::<String>());
	}
}

fn main() {
    std::thread::Builder::new()
        .stack_size(128 * 1024 * 1024)
        .spawn(|| solve()).unwrap()
        .join().unwrap();
}

0