結果

問題 No.1195 数え上げを愛したい（文字列編）
ユーザー 沙耶花沙耶花
提出日時 2020-08-22 17:29:23
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 499 ms / 3,000 ms
コード長 3,063 bytes
コンパイル時間 2,529 ms
コンパイル使用メモリ 217,392 KB
実行使用メモリ 23,696 KB
最終ジャッジ日時 2024-10-15 11:20:01
合計ジャッジ時間 11,321 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 493 ms
21,956 KB
testcase_01 AC 488 ms
23,696 KB
testcase_02 AC 499 ms
22,716 KB
testcase_03 AC 123 ms
14,752 KB
testcase_04 AC 137 ms
14,968 KB
testcase_05 AC 35 ms
13,764 KB
testcase_06 AC 23 ms
11,520 KB
testcase_07 AC 22 ms
11,520 KB
testcase_08 AC 112 ms
13,048 KB
testcase_09 AC 490 ms
22,352 KB
testcase_10 AC 239 ms
16,852 KB
testcase_11 AC 488 ms
21,752 KB
testcase_12 AC 476 ms
22,160 KB
testcase_13 AC 438 ms
17,712 KB
testcase_14 AC 239 ms
16,548 KB
testcase_15 AC 243 ms
16,732 KB
testcase_16 AC 248 ms
16,624 KB
testcase_17 AC 113 ms
13,224 KB
testcase_18 AC 487 ms
22,092 KB
testcase_19 AC 480 ms
22,600 KB
testcase_20 AC 415 ms
17,892 KB
testcase_21 AC 483 ms
21,868 KB
testcase_22 AC 259 ms
17,312 KB
testcase_23 AC 23 ms
11,648 KB
testcase_24 AC 22 ms
11,648 KB
testcase_25 AC 22 ms
11,776 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define modulo 998244353
#define mod(mod_x) ((((long long)mod_x+modulo))%modulo)
#define Inf 1000000005


int beki(long long a,long long b,int M = modulo){
	int x = 1;
	while(b!=0){
		if(b&1){
			x=((long long)x*a)%M;
		}
		a=((long long)a*a)%M;
		b>>=1;
	}
	return x;
}


int gyakugen(int a){
	return beki(a,modulo-2);
}

struct combi{
	deque<int> kaijou;
	deque<int> kaijou_;
	
	combi(int n){
		kaijou.push_back(1);
		for(int i=1;i<=n;i++){
			kaijou.push_back(mod(kaijou[i-1]*i));
		}
		
		int b=gyakugen(kaijou[n]);
		
		kaijou_.push_front(b);
		for(int i=1;i<=n;i++){
			int k=n+1-i;
			kaijou_.push_front(mod(kaijou_[0]*k));
		}
	}
	
	int combination(int n,int r){
		if(r>n)return 0;
		int a = mod(kaijou[n]*kaijou_[r]);
		a=mod(a*kaijou_[n-r]);
		return a;
	}
	
	int junretsu(int a,int b){
		int x = mod(kaijou_[a]*kaijou_[b]);
		x=mod(x*kaijou[a+b]);
		return x;
	}
	
	int catalan(int n){
		return mod(combination(2*n,n)*gyakugen(n+1));
	}
	
};

void NTT(vector<int> &A){
	int n = A.size();

	vector<int> tmp(n,0);
	int mask = n-1;
	
	for(int i=n>>1;i>=1;i>>=1){
		int r = (modulo-1)/n;
		r *= i;

		int w = beki(3,r);
		
		int now = 1;
	
		for(int j=0;j<n;j+=i){
			for(int k=0;k<i;k++){
				tmp[j+k] = mod(A[((j<<1)&mask)|k] + mod(now*A[(((j << 1)|i)&mask)|k]));
			}
			now = mod(now * w);
		}
		swap(A,tmp);
	}
	
	int num;
	for(int i=0;true;i++){
		if((n>>i)&1){
			num = i;
			break;
		}
	}

	if(num&1){
		swap(A,tmp);
		for(int i=0;i<n;i++)A[i] = tmp[i];
	}
	
	return;
}

vector<int> do_NTT(vector<int> A,vector<int> B){
	
	int n = A.size()+B.size()-1;
	
	if(min(A.size(),B.size())<=150){
		vector<int> ret(n,0);
		for(int i=0;i<A.size();i++){
			for(int j=0;j<B.size();j++){
				ret[i+j] = mod(ret[i+j] + mod(A[i]*B[j]));
			}
		}
		return ret;
	}
	
	
	for(int i=0;true;i++){
		if((1<<i)>=n){
			n = (1<<i);
			break;
		}
	}
	
	A.resize(n),B.resize(n);
	
	bool f = A==B;
	
	NTT(A);
	if(f)B=A;
	else NTT(B);
	
	int rev = gyakugen(n);
	
	vector<int> c(n);
	for(int i=0;i<n;i++)c[i] = mod(mod(A[i]*B[i])*rev);
	reverse(c.begin()+1,c.end());
	
	
	NTT(c);
	
	return c;
	
}

int main(){
	
	vector<int> cnt(26,0);
	string S;
	cin>>S;
	for(int i=0;i<S.size();i++){
		cnt[S[i]-'a']++;
	}
	combi C(1000000);
	vector<vector<int>> now;
	for(int i=0;i<26;i++){
		if(cnt[i]==0)continue;
		vector<int> x(cnt[i]+1,1);
		now.push_back(x);
	}
	
	while(now.size()!=1){
		vector<vector<int>> New;
		for(int i=0;i<now.size();i+=2){
			if(i+1==now.size()){
				New.push_back(now[i]);
				break;
			}
			
			for(int j=0;j<now[i].size();j++){
				now[i][j] = mod(now[i][j] * C.kaijou_[j]);
			}
			for(int j=0;j<now[i+1].size();j++){
				now[i+1][j] = mod(now[i+1][j] * C.kaijou_[j]);
			}
			
			vector<int> X = do_NTT(now[i],now[i+1]);
			for(int j=0;j<X.size();j++){
				X[j] = mod(X[j] * C.kaijou[j]);
			}
			while(X.size()>0&&X.back()==0)X.pop_back();
			New.push_back(X);
		}
		
		swap(now,New);
	}
	
	int ans  =0;
	for(int i=1;i<now[0].size();i++){
		ans = mod(ans + now[0][i]);
	}
	
	cout<<ans<<endl;

	
	return 0;
}
0