CS 400 Base K Xor 思维(数位)

题意:若a=20,b=14.当进制k=3时,a=202,b=112, a,b的digit sum a+b=011= 4 
digit sum为每位相加 舍去进位
Q次询问,每次给出[a,b,k]问a+(a+1)+..b的digit sum为多少? Q<=5e4,a,b<=1e8, 2<=k<=10


a,b,k={1,5,2} ans=1.
算[0,n] ans=[0,b]-[0,a-1].
每位都是独立计算的.当数字x在某一位出现kp次.则该位和为0(进位为px,结果为0).
容易发现[0,n]的最后一位 是循环出现的 0,1,2...k-1,0,1,2..k-1,....  最后一位的最终结果为 [ (k-1)*k/2 * n/k + (n%k+1)*(n%k)/2 ] %k

第p位也是有循环的 只不过是每个数字出现k^p后变到下一位. 
因为最后结果是要mod k的,所以只要考虑第p位最后一次出现的是那个数字 * 出现次数

第p位最后出现的显然为n的第p位,设为d,最后结果为(d*(n%k^p +1))%k 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll,ll> ii;
const int N=1e5+5;
ll a,b,k;
vector<ll> calc(ll n,ll k)
{
    ll last=((n/k)*(((k-1)*k)/2) + (n%k+1)*(n%k)/2 )%k;
    vector<ll> res;
    res.push_back(last);
    ll p=1,m=n;
    while(m)
    {
        m/=k;
        p=p*k;
        ll digit=m%k,fre=n%p+1;
        res.push_back((digit*fre)%k);
    }
    return res;
}
vector<ll> solve(ll a,ll b,ll k)
{
    vector<ll> v1=calc(a-1,k);
    vector<ll> v2=calc(b,k);
    for(int i=0;i<max(v1.size(),v2.size());i++)
        v2[i]=(v2[i]-(i<v1.size()?v1[i]:0)+k)%k;
    return v2;
}
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        cin>>a>>b>>k;
        vector<ll> sum=solve(a,b,k);
        ll ans=0,p=1;
        for(int i=0;i<sum.size();i++)
        {
            ans+=sum[i]*p;
            p=p*k;
        }
        cout<<ans<<endl;
    }
    return 0;
}