牛客 k进制数

链接：https://ac.nowcoder.com/acm/contest/907/D
来源：牛客网
 

题目描述

对于k进制数x，定义d(x)为x的各数位的和的k进制表示，如果结果超过一位，则继续重复执行各数位求和操作，直至结果为1位。
比如说，在7进制下，d(35047)=d((3+5+0+4)7)=d(157)=d((1+5)7)=d(67)=6
定义x为幸运的，当且仅当d(x) = b；
现在给定k进制下的n个数位a1 a2 an，问其中有多少子串组成的数字是幸运的。

输入描述:

第一行三个整数k，b，n（2 <= k <= 1,000,000,000, 0 <= b < k, 1 <= n <= 100,000)。
第二行n个整数满足0 <= ai < k。

输出描述:

输出一个整数表示幸运的子串数。

示例1

输入

复制

10 5 6
3 2 0 5 6 1

输出

复制

5

说明

3 2
3 2 0
0 5
5
2 0 5 6 1
是幸运的。

示例2

输入

复制

7 6 4
3 5 0 4

输出

复制

1

示例3

输入

复制

257 0 3
0 0 256

输出

复制

3

题解：
由于这个题求的是一个字符串所有的子串有多少数字满足k进制下d(x)=b，这个数字很明显会满足一些性质。 考 虑这个每次转化的过程，每一次进位相当于把一个k转化成了一个1，也就是说，在数位和mod(k-1)的意义下， 转化前和转化后的数字是等价的，最终会成为(x-1)mod(k-1)+1然后就不能动了 如果b=0，那么显然只有这 个串的所有数字都为0(也就是说这个数字为0)才成立，否则一个串一定不能转化为0否则这个串的和一定在mod(k-1)的意义下和b(mod(k-1))等价，这里简单的前缀和或者维护一个偏置值，维护前面的和(用map啥的)都 可以做 所以总之按照b=0分类，然后分类计算一下即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 5;
int a[maxn];
map<int,int> mp;
int main()
{
    int k, b, n, pre = 1, res = 0;
    ll ans = 0;
    scanf("%d%d%d",&k,&b,&n);
    for(int i = 1; i <= n; i++){
        scanf("%d",&a[i]);
        if(a[i]) pre = i+1;
        else ans += i-pre+1;
    }
    if(!b){
        printf("%lld\n",ans);
        return 0;
    }
    if(b == k-1) ans = -ans, b = 0;
    else ans = 0;
    mp[0] = 1;
    for(int i = 1; i <= n; i++){
        (res += a[i]) %= k-1;
        ans += mp[(res-b+k-1)%(k-1)];
        mp[res]++;
    }
    printf("%lld\n",ans);
    return 0;
}