結果

問題 No.3143 Colorless Green Parentheses Sleep Furiously
ユーザー ArcAki
提出日時 2025-05-16 23:15:15
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 29 ms / 2,000 ms
コード長 2,848 bytes
コンパイル時間 12,346 ms
コンパイル使用メモリ 380,184 KB
実行使用メモリ 23,424 KB
最終ジャッジ日時 2025-05-17 00:36:01
合計ジャッジ時間 13,855 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

/*
     ><>   魚fishに癒されたい  ><>
        >><>   ><>          ><>     ><>
    ><>         ><>     >>><>><><<>
         ><>   ><>   ><>    ><>
            ><>   >><>    ^<>
     ><>     >><>>    ><>
        >><><>      ><>
 */
#[allow(unused_imports)]
use std::{
    convert::{Infallible, TryFrom, TryInto as _}, fmt::{self, Debug, Display, Formatter,},
    fs::{File}, hash::{Hash, Hasher}, iter::{Product, Sum}, marker::PhantomData,
    ops::{Add, AddAssign, Sub, SubAssign, Div, DivAssign, Mul, MulAssign, Neg, RangeBounds},
    str::FromStr, sync::{atomic::{self, AtomicU32, AtomicU64}, Once},
    collections::{*, btree_set::Range, btree_map::Range as BTreeRange}, mem::{swap},
    cmp::{self, Reverse, Ordering, Eq, PartialEq, PartialOrd},
    thread::LocalKey, f64::consts::PI, time::Instant, cell::RefCell,
    io::{self, stdin, Read, read_to_string, BufWriter, BufReader, stdout, Write},
};
#[allow(unused_imports)]
use proconio::{input, input_interactive, marker::{*}};
#[allow(unused_imports)]
//use rand::{thread_rng, Rng, seq::SliceRandom};
#[allow(unused_imports)]
//use ac_library::{*, ModInt998244353 as mint};

#[allow(dead_code)]
const INF: i64 = 1<<60;
#[allow(dead_code)]
const MOD: i64 = 998244353;
#[allow(dead_code)]
const D: [(usize, usize); 4] = [(1, 0), (0, 1), (!0, 0), (0, !0)];
#[allow(dead_code)]
const D2: [(usize, usize); 8] = [(1, 0), (1, 1), (0, 1), (!0, 1), (!0, 0), (!0, !0), (0, !0), (1, !0)];

//use proconio::fastout;
//#[fastout]
fn main() {
    input!{
        n: usize, k: usize,
        s: Chars,
    }
    let mut cnt = 0;
    let mut edge = vec![Vec::new(); n+1];
    let mut pre = vec![0];
    let mut num = 0;
    for &c in s.iter(){
        if c=='('{
            num += 1;
            edge[pre[pre.len()-1]].push(num);
            pre.push(num);
            cnt += 1;
        } else if cnt==0{
            println!("No"); return;
        } else {
            pre.pop();
            cnt -= 1;
        }
    }
    if cnt!=0{
        println!("No"); return;
    } else {
        let mut ans = Vec::new();
        let mut ac = 0;
        dfs(0, &edge, &mut ac, &mut ans);
        if k >= ac{
            println!("Yes");
            while k > ac{
                ans.push("+1".to_string());
                ac += 1;
            }
            println!("{}", ans.join(""));
        } else {
            println!("No");
        }
    }
}

fn dfs(p: usize, edge: &Vec<Vec<usize>>, ac: &mut usize, ans: &mut Vec<String>){
    for &nex in &edge[p]{
        ans.push("(".to_string());
        dfs(nex, edge, ac, ans);
        ans.push(")".to_string());
        ans.push("+".to_string());
    }
    if edge[p].is_empty(){
        ans.push("1+1".to_string());
        *ac += 2;
    } else {
        ans.pop();
    }
    if edge[p].len()==1{
        ans.push("+1".to_string());
        *ac += 1;
    }
}
0