除法 CCF 201709-5

原题

问题描述
　　小葱喜欢除法，所以他给了你N个数a1, a2, ⋯, aN，并且希望你执行M次操作，每次操作可能有以下两种：
　　给你三个数l, r, v，你需要将al, al+1, ⋯, ar之间所有v的倍数除以v。
　　给你两个数l, r，你需要回答al + al+1 + ⋯ + ar的值是多少。
输入格式
　　第一行两个整数N, M，代表数的个数和操作的次数。
　　接下来一行N个整数，代表N个数一开始的值。
　　接下来M行，每行代表依次操作。每行开始有一个整数opt。如果opt=1，那么接下来有三个数l, r, v，代表这次操作需要将第l个数到第r个数中v的倍数除以v；如果opt = 2，那么接下来有两个数l, r，代表你需要回答第l个数到第r个数的和。
输出格式
　　对于每一次的第二种操作，输出一行代表这次操作所询问的值。
样例输入
5 3
1 2 3 4 5
2 1 5
1 1 3 2
2 1 5
样例输出
15
14
评测用例规模与约定
　　对于30%的评测用例，1 ≤ N, M ≤ 1000；
　　对于另外20%的评测用例，第一种操作中一定有l = r；
　　对于另外20%的评测用例，第一种操作中一定有l = 1 , r = N；
　　对于100%的评测用例，1 ≤ N, M ≤ 105，0 ≤ a1, a2, ⋯, aN ≤ 106, 1 ≤ v ≤ 106, 1 ≤ l ≤ r ≤ N。

思路

树状数组，比较精简的数据结构，能实现区间查询和单点修改，用在本题可以拿满分，直接套模板吧。

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;

const ll maxn=1e5+5;
ll a[maxn];//存放输入数据
ll d[maxn];//树状数组
ll n;//叶子结点数目

ll lowbit(ll x)
{//返回二进制数的最后一位1
    return x&(-x);
}

ll query(ll x)
{//返回a[1]~a[x]的和
    ll res=0;
    while(x)
    {
        res+=d[x];
        x-=lowbit(x);
    }
    return res;
}

void add(ll x,ll v)
{//实现对a[x]+v的修改
    while(x<=n)
    {
        d[x]+=v;
        x+=lowbit(x);
    }
}

int main()
{
    ll op,i,j,l,r,m,w;
    while(cin>>n>>m)
    {
        for(i=1;i<=n;i++)
        {
            cin>>a[i];
            add(i,a[i]);
        }
        for(i=1;i<=m;i++)
        {
            cin>>op>>l>>r;
            if(op==2)
                cout<<query(r)-query(l-1)<<endl;
            else
            {
                cin>>w;
                if(w==1)
                    continue;
                while(l<=r)
                {
                    if(a[l]>=w&&a[l]%w==0)
                    {
                        add(l,-(a[l]-a[l]/w));
                        a[l]/=w;
                    }
                    l++;
                }
            }
        }
    }
    return 0;
}