链接:https://ac.nowcoder.com/acm/contest/331/F
来源:牛客网
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld
题目描述
小希拿到了一个很长的两行的字符矩阵,她想知道里面有多少个子矩阵的字符取出后通过某种排列可以构成一个回文串。
该字符矩阵内仅含有数字0-9。
输入描述:
第一行输入一个整数N,表示字符矩阵的长度为N。
随后两行,每行N个字符,表示字符矩阵。
1≤N≤10000001≤N≤1000000
输出描述:
输出一个整数,表示有多少个子矩阵的字符取出后通过某种排列可以构成一个回文串。
示例1
输入
复制
2
00
00
输出
复制
9
示例2
输入
复制
10
0313512342
1231345123
输出
复制
28
题意:见题面;
解法:回文串中的奇数个的字符最多有一个,所以可以分别维护第一行、第二行、第一二行的异或前缀和,依次记录每一位之前的前缀异或值为i的个数cnt1[i](第一行),cnt2[i](第二行),cnt[i](第一二行);
附代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
char s[3][1000005];
int prexor1[1000005],prexor2[1000005],prexor[1000005],n; //记录前缀异或和
ll cnt1[10005],cnt2[10005],cnt[10005],ans=0; //记录前缀异或值为i的个数
int main()
{
scanf("%d",&n);
scanf("%s",s[1]+1);scanf("%s",s[2]+1);
cnt1[0]=1;cnt2[0]=1;cnt[0]=1;
for(int i=1;i<=n;i++){
prexor1[i]=prexor1[i-1]^(1<<(s[1][i]-'0'));
prexor2[i]=prexor2[i-1]^(1<<(s[2][i]-'0'));
prexor[i]=prexor[i-1]^(1<<(s[1][i]-'0'))^(1<<(s[2][i]-'0'));
ans+=cnt1[prexor1[i]]+cnt2[prexor2[i]]+cnt[prexor[i]]; //查询前面的异或值和当前异或值相同的个数
//即矩形中的所有数均为偶数个
for(int j=0;j<=9;j++) ans+=cnt1[prexor1[i]^(1<<j)]+cnt2[prexor2[i]^(1<<j)]+cnt[prexor[i]^(1<<j)];
//查询前面有一个字符为奇数个的子矩形个数
cnt1[prexor1[i]]++;
cnt2[prexor2[i]]++;
cnt[prexor[i]]++;
}
printf("%lld\n",ans);
}