QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#758073 | #5051. Namomo Subsequence | _xqy_ | WA | 388ms | 564856kb | C++20 | 2.4kb | 2024-11-17 15:35:43 | 2024-11-17 15:35:45 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
using LL = long long;
constexpr int mod = 998244353;
void solve() {
string s;
cin >> s;
int n = s.size();
s = " " + s;
auto get = [&] (char c) {
if (c >= 'a' && c <= 'z') return c - 'a';
else if (c >= 'A' && c <= 'Z') return c - 'A' + 26;
else return c - '0' + 52;
};
vector<int> t(n + 1);
for (int i = 1 ; i <= n ; ++i) {
t[i] = get(s[i]);
}
constexpr int M = 62;
vector<int> cnt(M) , sum(n + 1);
vector<vector<int>> pos(M);
vector<vector<int>> Pre(n + 1 , vector<int>(M));
vector<vector<int>> lst(n + 1 , vector<int>(M));
for (int i = 1 ; i <= n ; ++i) {
for (int j = 0 ; j < M ; ++j) {
Pre[i][j] = Pre[i - 1][j];
lst[i][j] = lst[i - 1][j];
}
Pre[i][t[i]] += 1;
pos[t[i]].emplace_back(i);
LL a1 = 1LL * cnt[t[i]] * (cnt[t[i]] - 1) / 2;
cnt[t[i]] += 1;
LL a2 = 1LL * cnt[t[i]] * (cnt[t[i]] - 1) / 2;
a2 -= a1;
sum[i] = (a2 + sum[i - 1]) % mod;
lst[i][t[i]] = (a2 + lst[i][t[i]]) % mod;
}
int ans = 0;
for (int i = 0 ; i < M ; ++i) {
if (pos[i].empty()) continue;
int m1 = pos[i].size() , k1 = m1 - 1;
vector<int> dp(m1);
int cnt_1_0 = 0;
for (int j = 0 ; j < M ; ++j) {
if (i == j || pos[j].empty()) continue;
int cnt_0_0 = 0 , cnt_0_1 = 0;
int m2 = pos[j].size() , k2 = m2 - 1;
while (k1 >= 0 || k2 >= 0) {
if (k1 >= 0 && k2 >= 0) {
if (pos[i][k1] > pos[j][k2]) {
cnt_1_0 += cnt_0_0;
if (cnt_1_0 >= mod) cnt_1_0 -= mod;
dp[k1] = cnt_0_1;
k1 --;
} else {
cnt_0_0 += 1 , cnt_0_1 += cnt_1_0;
if (cnt_0_1 >= mod) cnt_0_1 -= mod;
k2 --;
}
} else if (k1 >= 0) {
cnt_1_0 += cnt_0_0;
if (cnt_1_0 >= mod) cnt_1_0 -= mod;
dp[k1] = cnt_0_1;
k1 --;
} else {
break;
}
}
for (int k = 0 ; k < m1 ; ++k) {
int p = pos[i][k];
LL a = 1LL * (p - 1) - Pre[p - 1][i] - Pre[p - 1][j];
a = a * (a - 1) / 2;
LL b = 1LL * lst[p - 1][i] + 1LL * lst[p - 1][j] - 1LL * sum[p - 1];
a = (a + b) * dp[k] % mod;
ans += a;
if (ans >= mod) ans -= mod;
if (ans < 0) ans += mod;
dp[k] = 0;
}
cnt_1_0 = 0 , k1 = m1 - 1;
}
}
cout << ans << "\n";
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int t = 1;
while (t--) {
solve();
}
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3820kb
input:
wohaha
output:
1
result:
ok 1 number(s): "1"
Test #2:
score: 0
Accepted
time: 0ms
memory: 3600kb
input:
momomo
output:
0
result:
ok 1 number(s): "0"
Test #3:
score: 0
Accepted
time: 0ms
memory: 3844kb
input:
gshfd1jkhaRaadfglkjerVcvuy0gf
output:
73
result:
ok 1 number(s): "73"
Test #4:
score: 0
Accepted
time: 0ms
memory: 3548kb
input:
retiredMiFaFa0v0
output:
33
result:
ok 1 number(s): "33"
Test #5:
score: -100
Wrong Answer
time: 388ms
memory: 564856kb
input:
bcdccccacccabdbdaddcabddbaccaaaaaabaccbbbcbbddabcbccabacdacbcbabccbcbddcdcbcaaadddddccdbabaabcbbcaaadadacdaadbdccbddddabcbaaddbcadadcbcbaaccacabdababaabdccadaddacdcacdaabbadadaddbbcccbcddaccaadbbcaaccccdcacbdbdddbaccaacbcaccaaabccdadddbaabdbcaaccacdcdcbcdddacbcacdbbbdccdddccccabdbacddacbaacbbcaccdcd...
output:
667480152
result:
wrong answer 1st numbers differ - expected: '587599316', found: '667480152'