QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#793828#5051. Namomo SubsequenceLJY_ljyWA 2270ms760112kbC++143.6kb2024-11-30 02:19:142024-11-30 02:19:14

Judging History

你现在查看的是最新测评结果

  • [2024-11-30 02:19:14]
  • 评测
  • 测评结果:WA
  • 用时:2270ms
  • 内存:760112kb
  • [2024-11-30 02:19:14]
  • 提交

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 = 65; // 'a' - 'z', 'A' - 'Z', '0' - '9' 

char str[MAXN];
int 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 = ((1ll * cnt[I][u] * I) % MOD - cnt2[u][cnt[I][u] - 1] - 1ll * cnt[I][u] + 2 * MOD) % MOD; //DXC, X任意 
				 		long long ans7 = ((1ll * cnt[I][u]) * 1ll * (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: 4092kb

input:

wohaha

output:

1

result:

ok 1 number(s): "1"

Test #2:

score: 0
Accepted
time: 1ms
memory: 4016kb

input:

momomo

output:

0

result:

ok 1 number(s): "0"

Test #3:

score: 0
Accepted
time: 1ms
memory: 6088kb

input:

gshfd1jkhaRaadfglkjerVcvuy0gf

output:

73

result:

ok 1 number(s): "73"

Test #4:

score: 0
Accepted
time: 1ms
memory: 4072kb

input:

retiredMiFaFa0v0

output:

33

result:

ok 1 number(s): "33"

Test #5:

score: -100
Wrong Answer
time: 2270ms
memory: 760112kb

input:

bcdccccacccabdbdaddcabddbaccaaaaaabaccbbbcbbddabcbccabacdacbcbabccbcbddcdcbcaaadddddccdbabaabcbbcaaadadacdaadbdccbddddabcbaaddbcadadcbcbaaccacabdababaabdccadaddacdcacdaabbadadaddbbcccbcddaccaadbbcaaccccdcacbdbdddbaccaacbcaccaaabccdadddbaabdbcaaccacdcdcbcdddacbcacdbbbdccdddccccabdbacddacbaacbbcaccdcd...

output:

237012645

result:

wrong answer 1st numbers differ - expected: '587599316', found: '237012645'