zoukankan      html  css  js  c++  java
  • 树状数组区间修改and查询和

    在差分数组上稍加改变,就可以实现这个骚操作

    首先我们先来看一看普通的树状数组(基于差分)怎么暴力的求解区间和就是询问区间长度次和

    (sum^{i=1}_{len}sum^{j=1}_{i}base[j])

    base为原数列

    以上便是暴力求解,然后我们可以发现(base[i])被加了(p-i+1)

    于是乎,我们就可以改写上式成为下式

    ((len+1)sum^{i=1}_{len}-sum^{i=1}_{len}(base[i]*i))

    我们用一个树状数组维护((len+1)sum^{i=1}_{len}),另一个维护(sum^{i=1}_{len}(base[i]*i))就可以了

    #include<cstdio>
    #include<iostream>
    #include<algorithm>
    using namespace std;
    long long base_1[101000];
    long long base_2[101000];
    int len;
    void insert(long long val,int pos)
    {
    	int now=pos;
    	while(now<=len)
    	{
    		base_1[now]+=val;
    		base_2[now]+=val*pos;
    		now+=now&(-now);
    	}
    	return ;
    }
    long long sum(int pos)
    {
    	long long res1=0,res2=0,now=pos;
    	while(now)
    	{
    		res1+=base_1[now];
    		res2+=base_2[now];
    		now-=now&(-now);
    	}
    	return res1*(pos+1)-res2;
    }
    void updata(long long val,int pos)
    {
    	int now=pos;
    	while(now<=len)
    	{
    		base_1[now]+=val;
    		base_2[now]+=val*pos;
    		now+=now&(-now);
    	}
    	return ;
    }
    long long check(int l,int r)
    {
    	return sum(r)-sum(l-1);
    }
    int main()
    {
    	int n,m;
    	scanf("%d%d",&n,&m);
    	len=n;
    	long long pa,pb=0;
    	for(int i=1;i<=n;i++)
    	{
    		scanf("%lld",&pa);
    		updata(pa-pb,i);
    		pb=pa;
    	}
    	int a,b,c;
    	for(int i=1;i<=m;i++)
    	{
    		scanf("%d%d%d",&a,&b,&c);
    		if(a==1)
    		{
    			scanf("%lld",&pa);
    			updata(pa,b);
    			updata(-pa,c+1);
    		}
    		else
    		{
    			printf("%lld
    ",check(b,c));
    		}
    	}
    }
    
  • 相关阅读:
    周总结6
    河北省重大需求征集系统每日进度8
    《代码整洁之道》读书笔记3
    周总结5
    河北省重大需求征集系统每日进度7
    mybatis 关联查询多个类型相同对象
    thymeleaf 接受字符串类型model值作为js方法参数报错500
    毕业设计(2.17-2.23)周工作计划
    毕业设计记录——2.16
    毕业设计记录——2.15
  • 原文地址:https://www.cnblogs.com/Lance1ot/p/9214279.html
Copyright © 2011-2022 走看看