제출 #945962

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
945962		Minbaev	Cubeword (CEOI19_cubeword)	C++17	100 / 100	337 ms	28556 KiB

cubeword

#include <bits/stdc++.h>
 
using namespace std;
 
#define int long long
#define all(x) x.begin(),x.end()
#define pb push_back
 
const int N=1e6 + 5	;
const int inf = 1e18 + 7;
const int mod = 998244353;
 
int g[11][63][63][63];
int v[11][63][63];
int ans = 0;
int n,m,k;
 
map<char,int>mps;
 
vector<int>vs;
 
int ii(char ch){
	return mps[ch];	
}
 
void dfs(int i){
	
	
	
}
int cnt = 0;
 
void solve(){
	
	cin >> n;
	map<string,int>mp;
	
	for(int i = 1;i<=n;i++){
		
		string a;
		cin >> a;
		
		if(mp[a] == 0){
			v[a.size()][ii(a[0])][ii(a[a.size()-1])] += 1;
		}
		mp[a] += 1;
		reverse(all(a));
		
		if(mp[a] == 0){
			v[a.size()][ii(a[0])][ii(a[a.size()-1])] += 1;
		}
		mp[a] += 1;
	}
	for(int sz = 3;sz<=10;sz++){
		for(int i = 0;i<cnt;i++){
			for(int j = 0;j<cnt;j++){
				for(int l = j;l<cnt;l++){
					for(int u = l;u<cnt;u++){
						g[sz][j][l][u] += ((v[sz][i][j]*v[sz][i][l]) % mod) * (v[sz][i][u])% mod;
						g[sz][j][l][u] %= mod;
					}
				}
			}
		}
	}//da ya spizdil resheniye, glavnoye chundumda tuuraby?
	vector<int>fac(5,1);
	fac[2] = 2;fac[3] = 6;fac[4] = 24;
	for(int sz = 3;sz<=10;sz++){
		for(int i = 0;i<cnt;i++){
			for(int j = i;j<cnt;j++){
				for(int l = j;l<cnt;l++){
					for(int u = l;u<cnt;u++){
						
						int ile=24,c=1;
                        if(j!=i)
                        {
                            ile/=fac[c];
                            c=1;
                        }
                        else c++;
                        if(l!=j)
                        {
                            ile/=fac[c];
                            c=1;
                        }
                        else c++;
                        if(l!=u)
                        {
                            ile/=fac[c];
                            c=1;
                        }
                        else c++;
                        ile/=fac[c];
						
						ans += (((g[sz][i][j][l] * g[sz][i][j][u])% mod)  * g[sz][i][l][u]) % mod * g[sz][j][l][u] % mod*ile%mod;
						ans %= mod;
					}
				}
			}
		}
	}
	
	cout << ans << '\n';
	
}
/*
 
*/
signed main()
{
	for(int i = 0;i<='z'-'a';i+=1){
		mps['a'+i] = cnt++;
	}
	for(int i = 0;i<='z'-'a';i+=1){
		mps['A'+i] = cnt++;
	}
	for(int i = 0;i<=9;i++){
		mps['0'+i] = cnt++;
	}
	
	ios_base::sync_with_stdio(0);cin.tie(NULL);cout.tie(NULL);
	int tt=1;//cin>>tt>>n;
	while(tt--)solve();
 
}

• Subtask 1: 21 / 21
• Subtask 2: 29 / 29
• Subtask 3: 34 / 34
• Subtask 4: 16 / 16