AtCoder Beginner Contest 158 E Divisible Substring 記錄連續餘數和,並統計相同餘數和的個數

AtCoder Beginner Contest 158   比賽人數8029   codeforces比賽結束不久,就開打了,還好狀態還可以,就是網站太慢,浪費了20分鐘.

AtCoder Beginner Contest 158 E Divisible Substring  記錄連續餘數和,並統計相同餘數和的個數

總目錄詳見https://blog.csdn.net/mrcrack/article/details/104454762

在線測評地址https://atcoder.jp/contests/abc158/tasks/abc158_e

思路同https://blog.csdn.net/spnooyseed/article/details/104730595/

abcd % p = ( a * 1000 % p + b * 100 % p + c * 10 % p + d * 1 % p ) % p

#include <stdio.h>
#define LL long long
char s[200010];
int cnt[10010];
LL ans;
int main(){
	int n,p,i,pow10=1,now=0;
	scanf("%d%d%s",&n,&p,s+1);
	if(p==2||p==5){
		for(i=n;i>=1;i--)//從數據的最低位開始處理
			if((s[i]-'0')%p==0)ans+=i;
	}else{//p!=2,p!=5
		cnt[0]=1;
		for(i=n;i>=1;i--){
			now=(now+(s[i]-'0')*pow10)%p;
			pow10=pow10*10%p;
			ans+=cnt[now];//處理到i位時,連續餘數和爲now的個數是cnt[now],當前餘數和now與之前餘數和now,作差爲０,那麼之間的餘數和爲0的數能找到cnt[now]個，這就是能被p整除的數量。
			cnt[now]++;
		}
	}
	printf("%lld\n",ans);
	return 0;
}