D - Three Days Ago (atcoder.jp)
这个题可以用到二维优化。首先这个题的题意是:
我们要找到符合条件的子字符串满足
1 <= l <= r <= N(字符串长度),然后这个子字符串可以通过重新排列来重新规划顺序,所以我们只需要让这个字符串中的字符数量相等就行了。
我们可以用到二维表示以及用到异或的一些性质。我们首先开一个 2 ^ 10 的数组 1 << 10 即可,其中的每一位都表示从 0 开始到 9 的一个数字。首先我们设置一个 now 变量,用来存我们当前的值,然后我们遍历这个字符串,对于每个字符,我们让 now 变量来与这个数的二进制表示进行异或,然后由于异或中如果两个数出现偶数次,那必定是 0 ,或者是 1 ,然后我们可以查找之前是否出现过这样的情况,这两种情况中的就一定是每一个数都出现偶数次,所以我们直接让答案加上 dp[now] 就行了, 然后这种情况又出现了一次,所以 dp[now] ++, 并且不要忘了 dp[0] = 1,因为也有可能会出现从一开始每一个数都出现两次,所以这种情况也是成立的。
代码如下:
#include <bits/stdc++.h>
#define int long long
using namespace std;
int dp[1 << 10];
signed main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
string s; cin >> s;
dp[0] ++;
int now = 0;
int ans = 0;
for (char c : s)
{
now ^= 1 << (c - '0');
ans += dp[now] ++;
}
cout << ans << "\n";
return 0;
}