CCF201709-5(树状数组单点更新区间求和)

来源:互联网 发布:网络连接错误 编辑:程序博客网 时间:2024/06/05 00:17

问题描述
  小葱喜欢除法,所以他给了你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。

这个题疯狂卡我细节时间,心态GG。
开始的时候有些傻X,忘了用数组存值,结果每次都要先查询,再更新,卡了一波时间。。。

#include <cstdio>long long tree[101024];int n,m,a[101024];int lowbit(int x){    return (x&(-x));}void update(int i,int val){    while(i<=n)    {        tree[i]+=val;        i+=lowbit(i);    }}long long getsum(int i){    long long sum=0;    while(i>0)    {        sum+=tree[i];        i-=lowbit(i);    }    return sum;}int main(){    scanf("%d %d",&n,&m);    for(int i=1;i<=n;++i)    {        scanf("%d",&a[i]);        update(i,a[i]);    }    while(m--)    {        int opt,l,r,w;        scanf("%d %d %d",&opt,&l,&r);        if(opt==2) {            printf("%lld\n",getsum(r)-getsum(l-1));        }        else        {            scanf("%d",&w);            if(w==1)                continue;            for(int i=l;i<=r;i++)            {                if(a[i]>=w&&a[i]%w==0)                {                    update(l,-(a[l]-a[l]/w));                    a[l]/=w;                }                ++l;            }        }    }    return 0;}
阅读全文
0 0
原创粉丝点击