QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#793276 | #5051. Namomo Subsequence | LJY_ljy | TL | 2ms | 7888kb | C++11 | 3.5kb | 2024-11-29 18:12:27 | 2024-11-29 18:12:30 |
Judging History
answer
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <string>
#include <vector>
using namespace std;
const long long MOD = 998244353;
const long long MAXN = 1e6 + 10;
const long long MAXM = 64; // 'a' - 'z', 'A' - 'Z', '0' - '9'
char str[MAXN];
long long cnt[MAXN][MAXM];
long long Sum[MAXN];
vector<int> Index[MAXM];
vector<long long> sum[MAXM][MAXM];
vector<long long> cnt2[MAXM], cnt3[MAXM][MAXM];
long long AA[MAXN] = {0};
long long Tran(char c) {
if ('a' <= c && c <= 'z') return c - 'a' + 1;
if ('A' <= c && c <= 'Z') return c - 'A' + 27;
return c - '0' + 53;
}
int main() {
//freopen("test.in", "r", stdin);
//freopen("std.out", "w", stdout);
cin >> str + 1;
long long n = strlen(str + 1);
for (long long i = 1; i <= n; i++) {
for (long long j = 1; j <= 62; j++)
cnt[i][j] = cnt[i - 1][j];
cnt[i][Tran(str[i])] = cnt[i - 1][Tran(str[i])] + 1;
Index[Tran(str[i])].push_back(i);
Sum[i] = ( Sum[i - 1] + (cnt[i][Tran(str[i])] * (cnt[i][Tran(str[i])] - 1) / 2) % MOD - ((cnt[i][Tran(str[i])] - 1) * (cnt[i][Tran(str[i])] - 2) / 2) % MOD + MOD) % MOD;
}
for (long long i = 1; i <= 62; i++) {
long long A = 0;
for (long long j = 0; j < Index[i].size(); j++) {
A = (A + Index[i][j]) % MOD;
cnt2[i].push_back(A);
}
}
for (long long u = 1; u <= 62; u++) {
for (long long i = 1; i <= 62; i++) {
long long A = 0;
for (long long j = 0; j < Index[i].size(); j++) {
A = (A + cnt[Index[i][j]][u]) % MOD;
cnt3[i][u].push_back(A);
}
}
}
for (long long i = 1; i <= 62; i++) { // C
for (long long u = 1; u <= 62; u++) { // D
long long ans = 0, Sumsum = 0;
if (!Index[u].empty()) {
for (long long x = Index[u].size() - 1; x >= 0; x--) {
ans = (ans + cnt[Index[u][x]][i]) % MOD;
Sumsum = (Sumsum + ( ans - ( (Index[u].size() - x) * cnt[Index[u][x]][i]) % MOD + MOD) % MOD) % MOD;
sum[u][i].push_back(Sumsum);
}
}
}
}
long long Cnt = 0;
for (long long i = 1; i <= 62; i++) {
for (long long j = 0; j < Index[i].size(); j++) {
long long I = 1ll * Index[i][j];
long long ans1 = ( ( (I - 1) * (I - 2) / 2) % MOD - Sum[I - 1] + MOD) % MOD; // ABC
long long ans2 = 0, ans3 = 0;
if (j > 0) {
ans2 = (cnt2[i][j - 1] - cnt3[i][i][j - 1] + MOD) % MOD; //XCC;
ans3 = ( (1ll * j * I) % MOD - cnt2[i][j - 1] + 2 * MOD - (j * j) % MOD + ((j - 1) * j / 2) % MOD + MOD) % MOD; //CXC
}
long long A = (ans1 - ans2 - ans3 + 2 * MOD) % MOD; // ABC
long long AAA = 0;
for (long long u = 1; u <= 62; u++) {
if (u != i) {
long long ans5 = 0, ans6 = 0;
if (cnt[I][u] >= 1) {
ans5 = (cnt2[u][cnt[I][u] - 1] - cnt3[u][i][cnt[I][u] - 1] - cnt3[u][u][cnt[I][u] - 1] + 2 * MOD) % MOD; // XDC
ans6 = ((cnt[I][u] * I) % MOD - cnt2[u][cnt[I][u] - 1] - cnt[I][u] + 2 * MOD) % MOD; //DXC, X任意
long long ans7 = ((cnt[I][u]) * (cnt[I][u] - 1) / 2) % MOD; // DDC
long long ans8 = 0;
if (j >= 1)
ans8 = cnt3[i][u][j - 1]; //DCC
ans6 = (ans6 - ans7 - ans8 + 2 * MOD) % MOD;
}
long long B = (A - ans5 - ans6 + 2 * MOD) % MOD;
if (cnt[n][u] != cnt[I][u]) {
Cnt = ( Cnt + (B * sum[u][i][cnt[n][u] - cnt[I][u] - 1]) % MOD ) % MOD;
// AAA = (AAA + (B * sum[u][i][cnt[n][u] - cnt[I][u] - 1]) % MOD ) % MOD;
}
}
}
// AA[I] = AAA;
}
}
// for (long long i = 1; i <= n; i++) cout << AA[i] << " ";
// cout << endl;
printf("%lld\n", Cnt);
//fclose(stdin);
//fclose(stdout);
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 7876kb
input:
wohaha
output:
1
result:
ok 1 number(s): "1"
Test #2:
score: 0
Accepted
time: 0ms
memory: 7852kb
input:
momomo
output:
0
result:
ok 1 number(s): "0"
Test #3:
score: 0
Accepted
time: 0ms
memory: 5888kb
input:
gshfd1jkhaRaadfglkjerVcvuy0gf
output:
73
result:
ok 1 number(s): "73"
Test #4:
score: 0
Accepted
time: 2ms
memory: 7888kb
input:
retiredMiFaFa0v0
output:
33
result:
ok 1 number(s): "33"
Test #5:
score: -100
Time Limit Exceeded
input:
bcdccccacccabdbdaddcabddbaccaaaaaabaccbbbcbbddabcbccabacdacbcbabccbcbddcdcbcaaadddddccdbabaabcbbcaaadadacdaadbdccbddddabcbaaddbcadadcbcbaaccacabdababaabdccadaddacdcacdaabbadadaddbbcccbcddaccaadbbcaaccccdcacbdbdddbaccaacbcaccaaabccdadddbaabdbcaaccacdcdcbcdddacbcacdbbbdccdddccccabdbacddacbaacbbcaccdcd...