#1085353 (Rust) No.3143 Colorless Green Parentheses Sleep Furiously

• 問題: No.3143 Colorless Green Parentheses Sleep Furiously
• ユーザー: Moss_Local
• 提出日時: 2025-05-16 22:14:36
• 言語: Rust (1.83.0 + proconio)
• 結果: WA
• 実行時間: -
• コード長: 5,658 bytes
• コンパイル時間: 12,616 ms
• コンパイル使用メモリ: 400,460 KB
• 実行使用メモリ: 8,392 KB
• 最終ジャッジ日時: 2025-05-17 00:30:01
• 合計ジャッジ時間: 15,572 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

ファイルパターン	結果
sample	AC * 3
other	AC * 47 WA * 2

コンパイルメッセージ

warning: variable does not need to be mutable
   --> src/main.rs:119:9
    |
119 |     let mut vec: Vec<i64> = read_vec();
    |         ----^^^
    |         |
    |         help: remove this `mut`
    |
    = note: `#[warn(unused_mut)]` on by default

warning: variable does not need to be mutable
   --> src/main.rs:125:9
    |
125 |     let mut vec: Vec<i64> = read_vec();
    |         ----^^^
    |         |
    |         help: remove this `mut`

warning: variable does not need to be mutable
   --> src/main.rs:130:9
    |
130 |     let mut vec: Vec<usize> = read_vec();
    |         ----^^^
    |         |
    |         help: remove this `mut`

warning: variable does not need to be mutable
   --> src/main.rs:136:9
    |
136 |     let mut vec: Vec<f64> = read_vec();
    |         ----^^^
    |         |
    |         help: remove this `mut`

warning: variable does not need to be mutable
   --> src/main.rs:141:9
    |
141 |     let mut vec: Vec<char> = read_vec();
    |         ----^^^
    |         |
    |         help: remove this `mut`

warning: variable does not need to be mutable
   --> src/main.rs:146:9
    |
146 |     let mut vec: Vec<usize> = read_vec();
    |         ----^^^
    |         |
    |         help: remove this `mut`

warning: variable does not need to be mutable
   --> src/main.rs:151:9
    |
151 |     let mut vec: Vec<i64> = read_vec();
    |         ----^^^
    |         |
    |         help: remove this `mut`

warning: variable does not need to be mutable
   --> src/main.rs:157:9
    |
157 |     let mut vec: Vec<usize> = read_vec();
    |         ----^^^
    |         |
    |         help: remove this `mut`

warning: value assigned to `o` is never read
   --> src/main.rs:230:13
    |
230 |             o = true;
    |             ^
    |
    = help: maybe it is overwritten before being read?
    = note: `#[warn(unused_assignments)]` on by default

warning: variable does not need to be mutable
   --> src/main.rs:225:24
    |
225 |     while le

ソースコード

// -*- coding:utf-8-unix -*-
// #![feature(map_first_last)]
#![allow(dead_code)]
#![allow(unused_imports)]
#![allow(unused_macros)]
// use core::num;
use std::cmp::*;
use std::fmt::*;
use std::hash::*;
use std::io::BufRead;
use std::iter::FromIterator;
use std::*;
use std::{cmp, collections, fmt, io, iter, ops, str};
const INF: i64 = 1223372036854775807;
const UINF: usize = INF as usize;
const LINF: i64 = 2147483647;
const INF128: i128 = 1223372036854775807000000000000;
const MOD1: i64 = 1000000007;
const MOD9: i64 = 998244353;
const MOD: i64 = MOD9;
// const MOD: i64 = MOD2;
const UMOD: usize = MOD as usize;
const M_PI: f64 = 3.14159265358979323846;

// use proconio::input;
// const MOD: i64 = INF;

use cmp::Ordering::*;
use std::collections::*;

use std::io::stdin;
use std::io::stdout;
use std::io::Write;

macro_rules! p {
    ($x:expr) => {
        //if expr
        println!("{}", $x);
    };
}

macro_rules! vp {
    // vector print sd1arate with space
    ($x:expr) => {
        println!(
            "{}",
            $x.iter()
                .map(|x| x.to_string())
                .collect::<Vec<_>>()
                .join(" ")
        );
    };
}

macro_rules! d {
    ($x:expr) => {
        d1rintln!("{:?}", $x);
    };
}
macro_rules! yn {
    ($val:expr) => {
        if $val {
            println!("Yes");
        } else {
            println!("No");
        }
    };
}

macro_rules! map{
    // declear btreemap
    ($($key:expr => $val:expr),*) => {
        {
            let mut map = ::std::collections::BTreeMap::new();
            $(
                map.insert($key, $val);
            )*
            map
        }
    };
}

macro_rules! set{
    // declear btreemap
    ($($key:expr),*) => {
        {
            let mut set = ::std::collections::BTreeSet::new();
            $(
                set.insert($key);
            )*
            set
        }
    };
}

//input output
#[allow(dead_code)]
fn read<T: std::str::FromStr>() -> T {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).ok();
    s.trim().parse().ok().unwrap()
}

#[allow(dead_code)]
fn read_vec<T: std::str::FromStr>() -> Vec<T> {
    read::<String>()
        .split_whitespace()
        .map(|e| e.parse().ok().unwrap())
        .collect()
}

#[allow(dead_code)]
fn read_mat<T: std::str::FromStr>(n: u32) -> Vec<Vec<T>> {
    (0..n).map(|_| read_vec()).collect()
}

#[allow(dead_code)]
fn readii() -> (i64, i64) {
    let mut vec: Vec<i64> = read_vec();
    (vec[0], vec[1])
}

#[allow(dead_code)]
fn readiii() -> (i64, i64, i64) {
    let mut vec: Vec<i64> = read_vec();
    (vec[0], vec[1], vec[2])
}
#[allow(dead_code)]
fn readuu() -> (usize, usize) {
    let mut vec: Vec<usize> = read_vec();
    (vec[0], vec[1])
}

#[allow(dead_code)]
fn readff() -> (f64, f64) {
    let mut vec: Vec<f64> = read_vec();
    (vec[0], vec[1])
}

fn readcc() -> (char, char) {
    let mut vec: Vec<char> = read_vec();
    (vec[0], vec[1])
}

fn readuuu() -> (usize, usize, usize) {
    let mut vec: Vec<usize> = read_vec();
    (vec[0], vec[1], vec[2])
}
#[allow(dead_code)]
fn readiiii() -> (i64, i64, i64, i64) {
    let mut vec: Vec<i64> = read_vec();
    (vec[0], vec[1], vec[2], vec[3])
}

#[allow(dead_code)]
fn readuuuu() -> (usize, usize, usize, usize) {
    let mut vec: Vec<usize> = read_vec();
    (vec[0], vec[1], vec[2], vec[3])
}

// フレーム構造体
struct Frame {
    node: usize,
    idx: usize,
    leaf_cnt: usize,
    children: Vec<usize>,
    total: usize,
    opened: bool,
}

fn check_valid_parentheses(s: &str) -> bool {
    let mut stack = vec![];
    for c in s.chars() {
        if c == '(' {
            stack.push(c);
        } else if c == ')' {
            if stack.is_empty() {
                return false;
            }
            stack.pop();
        }
    }
    stack.is_empty()
}
fn main() {
    let (n, k) = readuu();
    if n % 2 == 1 {
        println!("No");
        return;
    }
    let s = read::<String>();
    if !check_valid_parentheses(&s) {
        println!("No");
        return;
    }
    let mut ch = vec![vec![]];
    let mut stk = vec![0];
    for c in s.chars() {
        if c == '(' {
            let id = ch.len();
            ch.push(vec![]);
            ch[*stk.last().unwrap()].push(id);
            stk.push(id);
        } else {
            stk.pop();
        }
    }
    let n = ch.len();
    let mut need = vec![0; n];
    let mut sum = 0;
    for i in 1..n {
        let c = ch[i].len();
        let m = c.max(2) - c;
        need[i] = m;
        sum += m;
    }
    if sum > k {
        println!("No");
        return;
    }
    let mut leaf = need.clone();
    leaf[0] = k - sum;
    let mut out = String::new();
    let mut stk = vec![(0, 0, false)];
    while let Some((v, mut i, mut o)) = stk.pop() {
        if !o {
            if v != 0 {
                out.push('(');
            }
            o = true;
        }
        let t = leaf[v] + ch[v].len();
        if i < t {
            if i > 0 {
                out.push('+');
            }
            if i < leaf[v] {
                out.push('1');
                stk.push((v, i + 1, true));
            } else {
                let u = ch[v][i - leaf[v]];
                stk.push((v, i + 1, true));
                stk.push((u, 0, false));
            }
        } else {
            if v != 0 {
                out.push(')');
            }
        }
    }
    // cnt 1 of out
    let mut cnt = 0;
    for c in out.chars() {
        if c == '1' {
            cnt += 1;
        }
    }
    if (cnt as i64) != k as i64 {
        println!("No");
        return;
    }
    println!("Yes");
    println!("{}", out);
}