纪中暑假培训 Date 6 ：矩形

Description

给定一个由数字（0-9）构成的字符串s。我们可以由此定义出size(s) * size(s) 大
小的矩阵b，其中b[i][j] = s[i] * s[j]；请问在这个矩阵b中，有多少子矩形满足其中的b[i][j]的和为另一个给定的数字a。

Input
第一行一个整数a。
第二行字符串s。

Output
一个整数表示满足条件的子矩形数。

Sample Input
10
12345

Sample Output
6
【样例解释】
b 矩阵为：
01 02 03 04 05
02 04 06 08 10
03 06 09 12 15
04 08 12 16 20
05 10 15 20 25
和为 10 的子矩形有：
一、01 02 03 04
二、
01
02
03
04
三、04 06
四、
04
06
五、10
六、10
以上共六个。

Data Constraint
对 10%的输入数据：size(s)≤10
对30%的输入数据：size(s)≤100
对100%的输入数据：0 ≤a≤1000000000，size(s)≤4000

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
using namespace std;

int n,len;
int a[37000],cnt[37000];
long long ans;
char s[37000];

int main()
{
    scanf("%d%s",&n,s+1);
    len=strlen(s+1);
    for(int i=1;i<=len;i++)
        a[i]=a[i-1]+s[i]-'0';      
    for(int i=1;i<=len;i++)
        for(int j=i;j<=len;j++)
            cnt[a[j]-a[i-1]]++;    
    if(n!=0)
    {
        for(int i=1;i<=len;i++)
            for(int j=i;j<=len;j++)
            {
                int k=a[j]-a[i-1];
                if(k==0) continue;
                if(n%k||(n/k)>37000) continue;
                ans+=cnt[n/k];
            }
    }       
    else
    {
        for(int i=1;i<=len;i++)
            for(int j=i;j<=len;j++)
            {
                int k=a[j]-a[i-1];
                if(k==0) ans+=len*(len+1)/2;     
                else
                    ans+=cnt[0];
            }
    }
    printf("%d",ans);
    return 0;
}