CodeForces - 364A Matrix(思维+数学)

题目链接：点击查看

题目大意：给出一个长度为 n 的，只由十进制数字组成的字符串 s，可以构造出一个大小为 n * n 的矩阵，构造方法如下：b[ i ][ j ] = s[ i ] * s[ j ]，现在问有多少个子矩阵的权值和等于 sum

题目分析：虽然 n 比较小，但如果模拟出数组 b 然后再求二维前缀和的话就有点麻烦了，这个题的模型和昨天做的那个题差不多：CodeForces - 1425D

上面的那个题是需要将完全平方和拆开，而这个题恰恰相反，是需要将完全平方和合起来，考虑一个区间内的权值和：，这个是两层for循环遍历所有的元素然后加和，模仿完全平方公式，我们不难推出更简单的公式：

因为一个矩形可以视为两个线段，一个是 x 方向的，一个是 y 方向的，上面的公式的两个乘数分别与其对应

这样一来我们就可以预处理出 y 方向上的所有 “线段”，再枚举 x 方向的每个 “线段” ，然后去 y 方向上去找到对应的与其匹配即可

不过需要特判一下 sum = 0 的情况，下面考虑一下 0 的贡献，假设 s[ i ] == 0，那么会使得 b 数组的第 i 行和第 i 列的取值都是 0，一行会产生 n * ( n + 1 ) / 2 个满足条件的子矩阵，而第 i 行和第 i 列同时为 0 的话，就会产生 n * ( n + 1 ) / 2 * 2 = n * ( n + 1 ) 个子矩阵，不过需要注意一下，b[ i ][ i ] 这个格子被计算了两次，所以需要减去

那么假设字符串 s 中共有 cnt[ 0 ] 个 “线段” 的取值为 0，那么这些 “线段” 既有 x 方向上的，又有 y 方向上的，根据上面容斥的扩展，总共是有 cnt[ 0 ] * n * ( n + 1 ) - cnt[ 0 ] * cnt[ 0 ] 个满足条件的子矩阵

代码：
 

//#pragma GCC optimize(2)
//#pragma GCC optimize("Ofast","inline","-ffast-math")
//#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
#include<cassert>
#include<bitset>
using namespace std;
 
typedef long long LL;
 
typedef unsigned long long ull;

const int inf=0x3f3f3f3f;

const int N=1e6+100;

char s[N];

LL cnt[N];

int sum[N];

int main()
{
#ifndef ONLINE_JUDGE
//  freopen("data.ans.txt","r",stdin);
//  freopen("data.out.txt","w",stdout);
#endif
//  ios::sync_with_stdio(false);
	int a;
	scanf("%d",&a);
	scanf("%s",s+1);
	int n=strlen(s+1);
	for(int i=1;i<=n;i++)
		sum[i]=sum[i-1]+s[i]-'0';
	for(int i=1;i<=n;i++)
		for(int j=1;j<=i;j++)
			cnt[sum[i]-sum[j-1]]++;
	if(a==0)
	{
		printf("%lld\n",cnt[0]*n*(n+1)-cnt[0]*cnt[0]);
		return 0;
	}
	LL ans=0;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=i;j++)
		{
			int num=sum[i]-sum[j-1];
			if(num!=0&&a%num==0&&a/num<N)
				ans+=cnt[a/num];
		}
	printf("%lld\n",ans);












    return 0;
}