AtCoder Beginner Contest 158 E - Divisible Substring

传送门

题意:

给一个字符串(由数字‘0’~‘9’组成),一个素数p,问有多少个非空子串是p的倍数

思路:

x记录到当前mod p的值,如果前面有几个值等于x,就加上几,因为这几个中任意一个到当前位置他们之间的字符串mod p==0

p=2||p=5,只要该为mod p==0,则以该为结束的字符串都是p的倍数

代码:

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <vector>
#include <math.h>
#include <map>
#include <queue>
#include <set>
#include <stack>
#define pb push_back
#define lb lower_bound
#define ub upper_bound
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
typedef long long ll;
using namespace std;
const int MAXN=2e5+50;
const int inf=0x3f3f3f3f;
const int M=5000*4;
char ch[MAXN];
int cnt[MAXN];
int main()
{
	int n,p;
	cin>>n>>p;
	cin>>(ch+1);
	ll ans=0;
	if(p==2||p==5){
		for(int i=1;i<=n;i++){
			if((ch[i]-'0')%p==0)ans+=i;
		}
		cout<<ans<<endl;
		return 0;
	}
	int t=1,x=0;
	cnt[0]=1;
	for(int i=n;i>=1;i--){
		x=(x+(ch[i]-'0')*t)%p;
		
		ans+=cnt[x];
		cnt[x]++;

		t=t*10%p;
	}
	cout<<ans<<endl;
    return 0;
}
/*

*/

类似的一个题2019牛客暑期多校训练营(第四场)K number(思维)

发布了161 篇原创文章 · 获赞 15 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_44091178/article/details/104821972
今日推荐