QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#793824 | #5051. Namomo Subsequence | LJY_ljy | ML | 1ms | 8100kb | C++14 | 3.6kb | 2024-11-30 02:15:35 | 2024-11-30 02:15:36 |
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 = 70; // 'a' - 'z', 'A' - 'Z', '0' - '9'
char str[MAXN];
long long cnt[MAXN][MAXM];
long long Sum[MAXN];
vector<int> Index[MAXM];
vector<int> sum[MAXM][MAXM];
vector<int> cnt2[MAXM], cnt3[MAXM][MAXM];
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() {
scanf("%s", 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()) {
cnt3[u][i].push_back(0);
for (long long x = Index[u].size() - 1; x >= 0; x--) {
ans = (ans + cnt[Index[u][x]][i]) % MOD;
cnt3[u][i].push_back(ans);
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][Index[i].size()] - cnt3[i][i][Index[i].size() - j] + MOD) % MOD + 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
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][Index[u].size()] - cnt3[u][i][Index[u].size() - cnt[I][u]] + MOD) % MOD - (cnt3[u][u][Index[u].size()] - cnt3[u][u][Index[u].size() - cnt[I][u]] + MOD) % MOD + 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][Index[i].size()] - cnt3[i][u][Index[i].size() - j] + MOD) % MOD; //DCC
ans6 = (ans6 - ans7 - ans8 + 2 * MOD) % MOD;
}
long long B = (A - ans5 - ans6 + 2 * MOD) % MOD;
if (cnt[n][u] != cnt[I][u]) {
//cout << B << " " << sum[u][i][cnt[n][u] - cnt[I][u] - 1] << endl;
Cnt = ( Cnt + (B * sum[u][i][cnt[n][u] - cnt[I][u] - 1]) % MOD ) % MOD;
}
}
}
}
}
printf("%lld\n", Cnt);
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 6220kb
input:
wohaha
output:
1
result:
ok 1 number(s): "1"
Test #2:
score: 0
Accepted
time: 1ms
memory: 4068kb
input:
momomo
output:
0
result:
ok 1 number(s): "0"
Test #3:
score: 0
Accepted
time: 1ms
memory: 8100kb
input:
gshfd1jkhaRaadfglkjerVcvuy0gf
output:
73
result:
ok 1 number(s): "73"
Test #4:
score: 0
Accepted
time: 1ms
memory: 4180kb
input:
retiredMiFaFa0v0
output:
33
result:
ok 1 number(s): "33"
Test #5:
score: -100
Memory Limit Exceeded
input:
bcdccccacccabdbdaddcabddbaccaaaaaabaccbbbcbbddabcbccabacdacbcbabccbcbddcdcbcaaadddddccdbabaabcbbcaaadadacdaadbdccbddddabcbaaddbcadadcbcbaaccacabdababaabdccadaddacdcacdaabbadadaddbbcccbcddaccaadbbcaaccccdcacbdbdddbaccaacbcaccaaabccdadddbaabdbcaaccacdcdcbcdddacbcacdbbbdccdddccccabdbacddacbaacbbcaccdcd...
output:
587599316