oj.uz

Submission #658464

# Submission time Handle Problem Language Result Execution time Memory
658464 2022-11-13T09:29:55 Z mychecksedad Cubeword (CEOI19_cubeword) C++17
0 / 100
 168 ms 3724 KB 
/* Author : Mychecksdead */
#include<bits/stdc++.h>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
using namespace std;
typedef long long int ll;
typedef long double ld;
#define MOD1 (1000000000+7)
#define MOD (998244353)
#define PI 3.1415926535
#define pb push_back
#define setp() cout << setprecision(15)
#define all(x) x.begin(), x.end()
#define debug(x) cerr << #x << " is " << x << '\n';
const int N = 1e6+100, M = 1e5+10, F = 2147483646, K = 20;

int n, S = 12;
ll dp[100][100];
vector<string> s[15];
bool is_palindrome(string &a){
    bool ok = 1;

    for(int i = 0; i < a.length() / 2; ++i) ok &= a[i] == a[a.length() - i - 1];

    return ok;
}
void solve(){
    cin >> n;
    for(int i = 0; i < n; ++i){
        string a; cin >> a;
        for(char &c: a) if(!(c >= 'a' && c <= 'p')) c = 'p' + (c - 'A') + 1;
        s[a.length()].pb(a);
    }
    ll ans = 0;
    for(int i = 1; i <= 10; ++i){
        if(s[i].empty()) continue;
        ll x = 1;
        for(int j = 0; j < S; ++j) for(int k = 0; k < S; ++k) dp[j][k] = 0;
        for(auto a: s[i]){
            if(is_palindrome(a)){
                dp[a[0] - 'a'][a[0] - 'a']++;
            }else{
                dp[a[0] - 'a'][a.back() - 'a']++;
                dp[a.back() - 'a'][a[0] - 'a']++;
            }
        }
        for(int left = 0; left < S*S; ++left){
            int x = left % S, y = left / S;
            for(int right = 0; right < S*S; ++right){
                int z = right % S, w = right / S;

                ll p = (dp[x][y] * dp[z][w]) % MOD;

                ll P = 0;

                for(int middle = 0; middle < S*S; ++middle){
                    int k = middle % S, l = middle / S;

                    ll pp = (dp[k][l] * dp[x][k] * dp[y][l]) % MOD;
                    ll ppp = (dp[z][k] * dp[w][l] * pp) % MOD;
                    
                    P += ppp;
                }
                P %= MOD;
                (ans += p * P * P);
            }
        }
    }
    cout << ans;
}





int main(){
    cin.tie(0); ios::sync_with_stdio(0);
    int T = 1, aa;
    // cin >> T;aa=T;
    while(T--){
        // cout << "Case #" << aa-T << ": ";
        solve();
        cout << '\n';
    }
    return 0;
 
}

Compilation message

cubeword.cpp: In function 'bool is_palindrome(std::string&)':
cubeword.cpp:23:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   23 |     for(int i = 0; i < a.length() / 2; ++i) ok &= a[i] == a[a.length() - i - 1];
      |                    ~~^~~~~~~~~~~~~~~~
cubeword.cpp: In function 'void solve()':
cubeword.cpp:37:12: warning: unused variable 'x' [-Wunused-variable]
   37 |         ll x = 1;
      |            ^
cubeword.cpp: In function 'int main()':
cubeword.cpp:78:16: warning: unused variable 'aa' [-Wunused-variable]
   78 |     int T = 1, aa;
      |                ^~

Subtask #1
0 / 21.0

# Verdict Execution time Memory Grader output
1 Incorrect 168 ms 3724 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 29.0

# Verdict Execution time Memory Grader output
1 Incorrect 168 ms 3724 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 34.0

# Verdict Execution time Memory Grader output
1 Incorrect 168 ms 3724 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
0 / 16.0

# Verdict Execution time Memory Grader output
1 Incorrect 168 ms 3724 KB Output isn't correct
2 Halted 0 ms 0 KB -