Codeforces Round #213 (Div. 2) C Matrix

num【len】【i】表示长度为len,起点是j(终点是j+len-1)的和;

vis【state】代表该数字出现过几次。因为最大和《=9*4000,所以vis数组大小无需超过40000;

所以ans就为:

if(vis[i]&&a%i==0)
if(a/i<=40000)
ans+=vis[i]*vis[a/i];

复杂度为n*n/2;

特别注意当0的时候,使用容斥原理求得ans;


#include
#include
#include
using namespace std;
#define LL long long 
LL num[4010][4010];
LL vis[41000];
char str[4100];
int main()
{
	LL a;
	LL sum;
	cin>>a;
	sum=0;
	cin>>str;
	int n=strlen(str);
	for(int i=1;i<=n;i++)
	{
		num[1][i]=str[i-1]-'0';
		vis[num[1][i]]++;
		sum++;
	}
	for(int i=2;i<=n;i++)
		for(LL j=1;j<=n-i+1;j++)
		{
			num[i][j]=num[i-1][j]+str[i+j-2]-'0';
			vis[num[i][j]]++;
			sum++;
		}
	if(a==0)
	{
		cout<<(vis[0]*sum*2-vis[0]*vis[0])<


你可能感兴趣的:(杂类)