#648781 (C++14) No.1481 Rotation ABC

提出ソース

結果

問題	No.1481 Rotation ABC
ユーザー	ayoiyouyoeyo
提出日時	2021-04-16 21:53:58
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	9 ms / 2,000 ms
コード長	2,120 bytes
コンパイル時間	3,613 ms
コンパイル使用メモリ	175,992 KB
実行使用メモリ	5,376 KB
最終ジャッジ日時	2024-07-03 01:34:15
合計ジャッジ時間	4,367 ms
ジャッジサーバーID （参考情報）	judge2 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,376 KB
testcase_02	AC	2 ms 5,376 KB
testcase_03	AC	1 ms 5,376 KB
testcase_04	AC	1 ms 5,376 KB
testcase_05	AC	1 ms 5,376 KB
testcase_06	AC	2 ms 5,376 KB
testcase_07	AC	1 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	2 ms 5,376 KB
testcase_10	AC	2 ms 5,376 KB
testcase_11	AC	2 ms 5,376 KB
testcase_12	AC	2 ms 5,376 KB
testcase_13	AC	8 ms 5,376 KB
testcase_14	AC	8 ms 5,376 KB
testcase_15	AC	7 ms 5,376 KB
testcase_16	AC	7 ms 5,376 KB
testcase_17	AC	6 ms 5,376 KB
testcase_18	AC	9 ms 5,376 KB
testcase_19	AC	6 ms 5,376 KB
testcase_20	AC	6 ms 5,376 KB
testcase_21	AC	8 ms 5,376 KB
testcase_22	AC	5 ms 5,376 KB
testcase_23	AC	9 ms 5,376 KB
testcase_24	AC	9 ms 5,376 KB
testcase_25	AC	9 ms 5,376 KB
testcase_26	AC	9 ms 5,376 KB
testcase_27	AC	9 ms 5,376 KB
testcase_28	AC	4 ms 5,376 KB
testcase_29	AC	5 ms 5,376 KB
testcase_30	AC	4 ms 5,376 KB
testcase_31	AC	5 ms 5,376 KB
testcase_32	AC	5 ms 5,376 KB
testcase_33	AC	5 ms 5,376 KB
testcase_34	AC	5 ms 5,376 KB
testcase_35	AC	4 ms 5,376 KB
testcase_36	AC	5 ms 5,376 KB
testcase_37	AC	5 ms 5,376 KB
testcase_38	AC	5 ms 5,376 KB
testcase_39	AC	5 ms 5,376 KB
testcase_40	AC	5 ms 5,376 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <iostream>
#include <string>
#include <cmath>
#include<algorithm>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<iomanip>
#include<bitset>
#define _USE_MATH_DEFINES
#include <math.h>
#include <functional>
#include<complex>
#include<cassert>
#include<random>
using namespace std;
#include<atcoder/all>
using namespace atcoder;


#define rep(i,x) for(ll i=0;i<x;i++)
#define repn(i,x) for(ll i=1;i<=x;i++)

typedef long long ll;
typedef pair<ll, ll> P;
const ll INF = 1e17;
const ll MAX = 4000001;
const long double eps = 1E-14;

ll max(ll a, ll b) {
	if (a > b) { return a; }
	return b;
}

ll min(ll a, ll b) {
	if (a > b) { return b; }
	return a;
}

ll gcd(ll a, ll b) {
	if (b == 0) { return a; }
	if (a < b) { return gcd(b, a); }
	return gcd(b, a % b);
}

ll lcm(ll a, ll b) {
	return a / gcd(a, b) * b;
}

struct edge {
	ll id;
	ll fr;
	ll to;
	ll d;
};

using mint = modint;

typedef vector<ll> vll;
typedef vector <vector<ll>> vvll;
typedef vector<vector<vector<ll>>> vvvll;

typedef vector<mint> vmint;
typedef vector<vector<mint>> vvmint;
typedef vector<vector<vector<mint>>> vvvmint;

vmint f, finv, inv;

void cominit(ll N) {
	ll MOD = modint::mod();//デフォルトが998244353に注意

	f.assign(N + 1, 1);
	finv.assign(N + 1, 1);
	inv.assign(N + 1, 1);
	inv[1] = 1;

	repn(i, N) {
		f[i] = f[i - 1] * i;
		if (i > 1)inv[i] = -inv[MOD % i] * (MOD / i);
		finv[i] = finv[i - 1] * inv[i];
	}
}

mint com(ll a, ll b) {
	if (a < 0 || b < 0 || a < b) { return 0; }
	return f[a] * finv[b] * finv[a - b];
}


/////////////////////////////////////

bool ch(string S, string T) {
	ll t = 0;
	rep(i, S.size()) {
		if (S[i] == T[t]) { t++; }
		if (t == T.size()) { return true; }
	}

	return false;
}


int main() {
	ll N;
	cin >> N;
	string S;
	cin >> S;
	set<char> st;
	rep(i, N)st.insert(S[i]);
	if (st.size() < 3) { cout << 1 << endl; return 0; }
	if (!ch(S, "ABC") && !ch(S, "BCA") && !ch(S, "CAB")) { cout << 1 << endl; return 0; }

	cominit(N + 1);
	vll c(3);
	rep(i, N)c[S[i] - 'A']++;

	mint ans = com(N, c[0]) * com(c[1] + c[2], c[1]);
	ans -= N;
	cout << ans.val() << endl;
}

yukicoder

結果

テストケース

ソースコード