AtCoder Beginner Contest 295 -- D - Three Days Ago(数学)

题目如下：

题目链接

题解 or 思路：

我们可以发现满足题意的子串长度一定为偶数
且每个数组出现的次数也必须是偶数次(或 $0$ 次)

所以我们把注意放在奇偶性上
我们预处理每个位置 $i$， $1-i$ 位置每个数出现次数的奇偶性。

我们寻找已存在且相同奇偶性的数量
只有奇偶性相同，减完才是偶数(或 $0$)

在这里我们可以用 $map$ 来记录
加起来就是最终答案

时间复杂度控制在 $O(n)$

AC 代码如下：

/*
Make it simple and keep self stupid
author：Joanh_Lan
*/
#pragma GCC optimize(3)
#pragma GCC optimize("inline") // 如果比赛允许开编译器优化的话，可以默写这两段
#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <numeric>
#include <cstring>
#include <cmath>
#include <map>
#include <unordered_map>
#include <bitset>
#include <set>
#include <random>
#include <ctime>
#include <queue>
#include <stack>
#include <climits>
#define buff                     \
    ios::sync_with_stdio(false); \
    cin.tie(0);
#define int long long
#define ll long long
#define PII pair<int, int>
#define px first
#define py second
typedef std::mt19937 Random_mt19937;
Random_mt19937 rnd(time(0));
using namespace std;
const int mod = 1e9 + 7;
const int inf = 2147483647;
const int N = 500009;
//int Mod(int a,int mod){return (a%mod+mod)%mod;}
//int lowbit(int x){return x&-x;}//最低位1及其后面的0构成的数值
//int qmi(int a, int k, int p){int res = 1 % p;while (k){if (k & 1) res = Mod(res * a , p);a = Mod(a * a , p);k >>= 1;}return res;}
//int inv(int a,int mod){return qmi(a,mod-2,mod);}
//int lcm(int a,int b){return a*b/__gcd(a,b);}
char s[N];
int cnt[10][N];
map<string, int> mp;
int ans = 0;
void calc(int x)
{
	string str;
	for (int i = 0; i <= 9; i++)
	{
		int tt = cnt[i][x];
	    str = str + char(tt + '0');
	}
	ans += mp[str];
	mp[str]++;
}
void solve()
{
	cin >> s + 1;
	int len = strlen(s + 1);
	mp["0000000000"]++;
	for (int i = 1; i <= len; i++)
	{
		int tt = (s[i] - '0');
		for (int j = 0; j <= 9; j++)
			cnt[j][i] = (cnt[j][i - 1] + (j == tt)) % 2;
		string str = "";
		calc(i);
	}
	cout << ans << '\n';
}
signed main()
{
	buff;
	int _ = 1;
	// cin >> _;
	while (_--)
		solve();
}