結果

問題 No.3143 Colorless Green Parentheses Sleep Furiously
ユーザー kwm_t
提出日時 2025-05-16 22:30:43
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,104 bytes
コンパイル時間 3,505 ms
コンパイル使用メモリ 295,064 KB
実行使用メモリ 25,600 KB
最終ジャッジ日時 2025-05-17 00:32:49
合計ジャッジ時間 9,305 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21 WA * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
//#include <atcoder/all>
using namespace std;
//using namespace atcoder;
//using mint = modint1000000007;
//const int mod = 1000000007;
//using mint = modint998244353;
//const int mod = 998244353;
//const int INF = 1e9;
//const long long LINF = 1e18;
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define rep2(i,l,r)for(int i=(l);i<(r);++i)
#define rrep(i, n) for (int i = (n) - 1; i >= 0; --i)
#define rrep2(i,l,r)for(int i=(r) - 1;i>=(l);--i)
#define all(x) (x).begin(),(x).end()
#define allR(x) (x).rbegin(),(x).rend()
#define P pair<int,int>
template<typename A, typename B> inline bool chmax(A & a, const B & b) { if (a < b) { a = b; return true; } return false; }
template<typename A, typename B> inline bool chmin(A & a, const B & b) { if (a > b) { a = b; return true; } return false; }
bool chk(string s) {
	int n = s.size();
	int c = 0;
	rep(i, n) {
		if (s[i] == '(')c++;
		else c--;
		if (c < 0) return false;
	}
	return c == 0;
}
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int n, k;
	string s;
	cin >> n >> k >> s;
	if (!chk(s)) {
		cout << "No" << endl;
		return 0;
	}
	vector<int>g(n, -1);
	{
		stack<int>st;
		rep(i, n) {
			if (s[i] == '(')st.push(i);
			else {
				auto top = st.top();
				g[top] = i + 1;
				st.pop();
			}
		}
	}
	auto dfs = [&](auto &&self, int st, int ed, int type = 1)->tuple<string, bool, int> {
		//cout << st << " " << ed << " " << type << endl;
		vector<P>ps;
		if (type == 0) {
			vector<P>ps;
			ps.emplace_back(st, g[st]);
			{
				int tmp = g[st];
				while (tmp < ed) {
					ps.emplace_back(tmp, g[tmp]);
					tmp = g[tmp];
				}
			}
			string tmp;
			int sum = 0;
			for (auto[fi, se] : ps) {
				auto[r0, r1, r2] = self(self, fi, se, 1);
				//cout << r0 << " " << r1 << "_" << r2 << endl;
				if (!r1)return { "",false,0 };
				if (!tmp.empty())tmp += "+";
				tmp += r0;
				sum += r2;
			}
			if (sum == k) {
				return { tmp,true,k };
			}
			else if (sum + 2 == k) {
				tmp = "1+" + tmp + "+1";
				return { tmp,true,k };
			}
			else if (sum + 1 == k) {
				tmp = "1+" + tmp;
				return { tmp,true,k };
			}
			else {
				return { "",false,0 };
			}
		}
		else {
			if (st + 2 == ed) {
				return { "(1+1)",true,2 };
			}
			vector<P>ps;
			ps.emplace_back(st + 1, g[st + 1]);
			{
				int tmp = g[st + 1];
				while (tmp + 1 < ed) {
					ps.emplace_back(tmp, g[tmp]);
					tmp = g[tmp];
				}
			}
			if (ps.size() == 1) {
				auto[s0, b0, n0] = self(self, ps[0].first, ps[0].second);
				auto s = "(1+" + s0 + ")";
				auto b = b0;
				auto n = n0 + 1;
				return { s,b,n };
			}
			else if (ps.size() == 2) {
				auto[s0, b0, n0] = self(self, ps[0].first, ps[0].second);
				auto[s1, b1, n1] = self(self, ps[1].first, ps[1].second);
				auto s = "(" + s0 + "+" + s1 + ")";
				auto b = b0 & b1;
				auto n = n0 + n1;
				return { s,b,n };

			}
			else {
				return { "",false,0 };
			}
		}
	};
	auto[ss, b, sum] = dfs(dfs, 0, n, 0);
	//cout << ss << " " << b << " " << sum << endl;
	if (!b || sum != k)cout << "No" << endl;
	else {
		cout << "Yes" << endl;
		cout << ss << endl;
	}
	return 0;
}
0