結果

問題 No.563 超高速一人かるた large
ユーザー PulmnPulmn
提出日時 2017-08-18 22:51:18
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 1,840 bytes
コンパイル時間 734 ms
コンパイル使用メモリ 68,472 KB
実行使用メモリ 146,644 KB
最終ジャッジ日時 2024-10-14 15:00:26
合計ジャッジ時間 3,676 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 83 ms
102,784 KB
testcase_01 AC 92 ms
102,912 KB
testcase_02 AC 79 ms
102,656 KB
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 AC 109 ms
102,912 KB
testcase_13 AC 111 ms
102,784 KB
testcase_14 WA -
testcase_15 AC 108 ms
102,656 KB
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 AC 137 ms
146,644 KB
testcase_20 WA -
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In member function ‘void Trie::dfs(int, vl&, ll&)’:
main.cpp:46:21: warning: ‘S’ may be used uninitialized in this function [-Wmaybe-uninitialized]
   46 |                 int S;
      |                     ^

ソースコード

diff # 

#include <iostream>
#include <cassert>
#include <vector>
#include <string>
using namespace std;
typedef long long ll;
typedef vector<ll> vl;
typedef vector<vl> vvl;
const ll mod=1e9+7;

const ll M=2005;

ll C[M][M],P[M][M];

void Init(){
	C[0][0]=P[0][0]=1;
	for(int i=0;i<M;i++) for(int j=0;j<=i;j++) if(i||j){
		if(i>j) C[i][j]=C[i-1][j];
		if(i&&j) (C[i][j]+=C[i-1][j-1])%=mod;
		if(j) P[i][j]=P[i-1][j-1]*i%mod;
		else P[i][j]=1;
	}
}

int n;

class Trie{
	private:
	const int MAX=210000;
	int ID=1;
	vvl date;
	vl a,u,dp;
	void Fix(vl& dp,ll h,int v){
		if(u[v]==1){
			dp[1]=1;
			return;
		}
		for(int i=1;i<=u[v];i++) (dp[i]+=P[u[v]][i]%mod*h)%=mod;
	}
	void dfs(int v,vl& dp,ll& h){
		if(a[v]==1){
			for(int i=0;i<27;i++) if(date[v][i]!=-1) dfs(date[v][i],dp,h);
			h++;
			return;
		}
		int S;
		bool b=1;
		for(int i=0;i<27;i++){
			int v_=date[v][i];
			if(v_!=-1){
				vl DP(u[v_]+1);
				ll H=0;
				dfs(v_,DP,H);
				Fix(DP,H,v_);
				if(b){
					dp=DP;
					b=0;
					S=u[v_];
				}
				else{
					vl dp_(S+u[v_]+1);
					for(int j=0;j<=S;j++) for(int k=0;k<=u[v_];k++) (dp_[j+k]+=(P[u[v_]][k]*dp[j]%mod+P[S][j]*DP[k]%mod)*C[j+k][j])%=mod;
					S+=u[v_];
					dp=dp_;
				}
			}
		}
		if(!a[v]) dp=vl(2);
		for(int i=1;i<=u[v];i++) (dp[i]+=(i-(i==u[v]?1:0))*P[u[v]][i])%=mod;
	}
	public:
	Trie(){
		date=vvl(MAX,vl(27,-1));
		a=u=vl(MAX);
	}
	void Update(string s){
		int S=s.size(),I=0;
		u[0]++;
		for(int i=0;i<S;i++){
			int c=s[i]-'a';
			if(date[I][c]==-1){
				date[I][c]=ID++;
				a[I]++;
			}
			I=date[I][c];
			u[I]++;
		}
	}
	void Solve(){
		Init();
		vl dp;
		ll h=0;
		dfs(0,dp,h);
		Fix(dp,h,0);
		for(int i=1;i<=n;i++) cout<<(dp[i]-(i-(i==n?1:0))*P[n][i]%mod+mod)%mod<<endl;
	}
};

int main(){
	cin>>n;
	Trie trie;
	for(int i=0;i<n;i++){
		string s;
		cin>>s;
		s+="{";
		trie.Update(s);
	}
	trie.Solve();
}
0