刷题记录（NC23054 华华开始学信息学）（树状数组）

NC23054    华华开始学信息学

题目链接

关键点：

对于d的操作，如果d很大的话，说明其计算因子的数目较少，就直接计算因子，然后相加，如果因子小，那么计算因子的操作就多，可以利用一个add数组，表示当前因子为x的+y

add[x] += y;

最后计算总数，先定义一个calc数组，用来计算树状数组从第一个到下标为x的区间和，即x的前缀和，因此先不计算add数组，计算出总和

ll sum = calc(y) - calc(x-1);

然后再从将add数组计算进去，如何计算[x,y]区间的add数组的影响呢，可以发现，比如对于以y为结尾的区间，我们可以用该区间内的最后一个数y对于add数组存在数的下标j相除，所得的数为以该最后数y为结尾的区间内，总共有因子j的总个数。

for (int j=1; j*j<=n; j++)
sum += ((y/j) - (x-1)/j)*add[j];

树状数组：

计算出x的形如00001000的形式

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

首先要将总数n表示成该形式，即包括n，且从第一个数开始的区间

while (lowbit(n) != n) n+=lowbit(n);

找数组里下标所有包含i因子的数 ，首先a[i]这个区间是包含i的最小下标区间，然后我们再找其他区间也包含i。

void Add(int i, int x)
{
    while (i<=n)//找所有有因子a[i]
    {
        a[i] += x;
        i += lowbit(i);//计算下一个有a[i]因子
    }
}

计算所有以下标第x数为结尾的区间和。

ll calc(int x)
{
    ll sum = 0;
    while (x>0)//找a[x]的所有因子
    {
        sum += a[x];
        x -= lowbit(x);//找x的因子
    }
    return sum;
}

完整代码：

# include <bits/stdc++.h>
using namespace std;
int n, m;
long long a[100000*2+10];
long long add[1000];
int lowbit(int x)
{
    return x&(-x);
}
void Add(int i, int x)
{
    while (i<=n)
    {
        a[i] += x;
        i += lowbit(i);
    }
}
long long calc(int x)
{
    long long sum = 0;
    while (x>0)
    {
        sum += a[x];
        x -= lowbit(x);
    }
    return sum;
}
int main()
{
    scanf("%d%d", &n, &m);
    while (lowbit(n) != n) n+=lowbit(n);
    for (int i=1; i<=m; i++)
    {
        int op, x, y;
        scanf("%d%d%d", &op, &x, &y);
        if (op == 1)
        {
            if (x*x <= n) add[x] += y;
            else
            {
                for (int j=x; j<=n; j+=x)
                    Add(j, y);
            }
        }
        else
        {
            long long sum = calc(y) - calc(x-1);
            for (int j=1; j*j<=n; j++)
            {
                sum += ((y/j) - (x-1)/j)*add[j];
            }
            printf("%lld\n", sum);
        }
    }
    
    
    return 0;
}