链接:
D-区间问题1_河南萌新联赛2024第(五)场:信息工程大学 (nowcoder.com)
题目描述:
Alice 有 n 个数,她可以对这 n 个数执行以下两种操作:
1. 将区间 [L,R] 上的所有数加上 d
2. 查询第 x 个数的值
输入描述:
第一行一个整数 n,表示数的个数,其中 n<=10^5;
第二行 n 个整数,表示 n 个数的初始值;
第三行一个整数 q ,表示操作数, q<=10^5;
接下来 q 行,第一个数表示操作类型,后面跟上对应的参数:
1 L R d,表示区间修改;
2 x,表示查询。
(过程数据保证在long long之内)
输出描述:
每一个操作2,对应一行输出,即查询的值。
示例1
输入
5 1 2 3 4 5 5 1 1 3 2 1 3 5 -1 2 3 1 2 4 5 2 4
输出
4 8
代码实现:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define lowbit(x) ((x)& -(x))
const int N=100010;
int n,m;
ll a[N];
ll tree[N];
void update(int x,ll d)
{
while(x<=n)
{
tree[x]+=d;
x+=lowbit(x);
}
}
ll sum(int x)
{
ll ans=0;
while(x>0)
{
ans+=tree[x];
x-=lowbit(x);
}
return ans;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
update(i,a[i]-a[i-1]);
}
cin>>m;
while(m--)
{
int op;
cin>>op;
if(op==1)
{
int L,R;
ll d;
cin>>L>>R>>d;
update(L,d);
update(R+1,-d);
}
else
{
int x;
cin>>x;
cout<<sum(x)<<endl;
}
}
return 0;
}