题目大意:给定多个字符串,把这些字符串随机两两合并后,形成的新字符串要求长度的偶数,并且前一半的数字和要等与后一半的数字和,问有多少种方案?
解题思路:我们定义一个四维数组,f[l][k][pre][post],意义为长度为L的字符串选中后K位,前面未选中的字符的数字和位pre,后面K位被选中的数字和为post的字符串数量 ,读入字符串后先对其进行处理,后面遍历数组直接分类讨论即可,复杂度为O(KN)
#include <bits/stdc++.h>
using namespace std;
#define int long long
int f[10][10][50][50],n,res;
string s[200010];
signed main ()
{
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>s[i];
int num=0,pre=0,len=s[i].size();
for(int j=0;j<len;j++)
{
num+=(s[i][j]-'0');
}
f[len][len][0][num]++;
for(int j=0;j<len;j++)
{
pre+=(s[i][j]-'0');
f[len][len-j-1][pre][num-pre]++;
}
}
for(int i=1;i<=n;i++)
{
int num=0,len=s[i].size();
for(int j=0;j<len;j++)
{
num+=(s[i][j]-'0');
}
if(len==1)
{
res+=f[1][1][0][num];
for(int j=0;j<=9;j++)
{
res+=f[3][2][j][j+num];
}
for(int j=0;j<=18;j++)
{
res+=f[5][3][j][j+num];
}
}else if(len==2)
{
res+=f[2][2][0][num];
for(int j=0;j<=9;j++)
{
res+=f[4][3][j][j+num];
}
}
else if(len==3)
{
res+=f[3][3][0][num];
int k=num-2*(s[i][2]-'0');
if(k>=0)
{
res+=f[1][1][0][k];
}
for(int j=0;j<=9;j++)
{
res+=f[5][4][j][j+num];
}
}else if(len==4)
{
res+=f[4][4][0][num];
int k=num-2*(s[i][3]-'0');
if(k>=0)
{
res+=f[2][2][0][k];
}
}else
{
res+=f[5][5][0][num];
int k=num-2*(s[i][3]-'0'+s[i][4]-'0');
if(k>=0)
{
res+=f[1][1][0][k];
}
k=num-2*(s[i][4]-'0');
if(k>=0)
{
res+=f[3][3][0][k];
}
}
}
cout<<res;
return 0;
}