2019ICPC 上海网络赛 L. Digit sum（二维树状数组+区间求和）

https://nanti.jisuanke.com/t/41422

 

 

 题目大意：

给出n和b，求1到n，各数在b进制下各位数之和的总和。

 

直接暴力模拟，TLE。。

没想到是要打表。。。还是太菜了。

 

 

 

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <string>
#include <math.h>
#include <algorithm>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <math.h>
const int INF=0x3f3f3f3f;
typedef long long LL;
const int mod=1e9+7;
const int maxn=1e6+10;
using namespace std;

int c[11][maxn];

int lowbit(int x)
{
    return x&(-x);
}

void Update(int f,int n,int val)
{
    while(n<maxn)
    {
        c[f][n]+=val;
        n+=lowbit(n);
    }
}

int Query(int f,int n)
{
    int sum=0;
    while(n>0)
    {
        sum+=c[f][n];
        n-=lowbit(n);
    }
    return sum;
}

void init()//对每个进制建立一个树状数组，将2到10进制的所有数的结果全计算出来
{
    for(int i=2;i<=10;i++)
    {
        for(int j=1;j<maxn;j++)
        {
            int sum=0;
            int t=j;
            do{
                sum+=(t%i);
                t/=i;
            }while(t);
            Update(i,j,sum);
        }
    }
}

int main()
{
    int T;
    scanf("%d",&T);
    init();//打表 
    for(int k=1;k<=T;k++)
    {
        int n,b;
        scanf("%d %d",&n,&b);
        printf("Case #%d: %d\n",k,Query(b,n));
    }
    return 0;
}