2018.8.18T1(线段树)
程序员文章站
2024-03-20 11:12:16
...
宝石(gem)
【题目描述】
有 n 座城市,编号为 1~n,第 i 座城市里宝石的交易价格为 ai。当
你经过第 i 座城市时,你可以以 ai 的价格购买或卖出一个宝石。在任
意时刻,你最多只能携带一个宝石。
有 m 次操作,操作分为两种:
(1) 给定l,r,询问依次经过编号为l~r的城市能获得的最大收益。
(2) 给定 l,r,x,y,将 al 至 ar 修改为首项为 x,公差为 y 的等差数
列。
【输入数据】
第一行两个整数 n,m,第二行 n 个整数 a1~an,接下来 m 行每行
第一个整数表示操作编号,接下来一些整数表示操作参数。
【输出数据】
对于每个操作 1,输出一行一个整数表示答案。
【样例输入】
6 5
3 2 1 2 3 4
1 1 5
2 1 3 1 1
1 2 6
1 1 6
1 3 4
【样例输出】
2
3
4
0
【数据范围】
对于 20%的数据,n,m<=1000。
对于另外 30%的数据,不存在操作 2。
对于又另外 20%的数据,操作 2 中 l=r。
对于又另外 20%的数据,操作 2 中 y=0。
对于 100%的数据,n,m<=200000,1<=l<=r<=n,1<=ai,x<=10^9,
|y|<=10^4
比较套路的裸题吧
直接用线段树维护差值
修改的时候相当于区间赋值和单点修改
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define rep(i,j,k) for(int i = j;i <= k;++i)
#define repp(i,j,k) for(int i = j;i >= k;--i)
#define rept(i,x) for(int i = linkk[x];i;i = e[i].n)
#define P pair<int,int>
#define ls root*2
#define rs root*2+1
const ll INF = 0x7fffffff;
int n , m;
int a[201000];
ll det[801000];
ll sum1[801000] , sum2[801000];
namespace fastIO{
#define BUF_SIZE 100000
#define OUT_SIZE 100000
bool IOerror = 0;
inline char nc(){
static char buf[BUF_SIZE],*p1 = buf+BUF_SIZE, *pend = buf+BUF_SIZE;
if(p1 == pend){
p1 = buf; pend = buf+fread(buf, 1, BUF_SIZE, stdin);
if(pend == p1){ IOerror = 1; return -1;}
}
return *p1++;
}
inline bool blank(char ch){return ch==' '||ch=='\n'||ch=='\r'||ch=='\t';}
inline void read(int &x){
bool sign = 0; char ch = nc(); x = 0;
for(; blank(ch); ch = nc());
if(IOerror)return;
if(ch == '-') sign = 1, ch = nc();
for(; ch >= '0' && ch <= '9'; ch = nc()) x = x*10+ch-'0';
if(sign) x = -x;
}
inline void read(ll &x){
bool sign = 0; char ch = nc(); x = 0;
for(; blank(ch); ch = nc());
if(IOerror) return;
if(ch == '-') sign = 1, ch = nc();
for(; ch >= '0' && ch <= '9'; ch = nc()) x = x*10+ch-'0';
if(sign) x = -x;
}
#undef OUT_SIZE
#undef BUF_SIZE
};
using namespace fastIO;
void update(int root)
{
sum2[root] = sum1[root] = 0;
sum2[root] += sum2[ls] > 0 ? sum2[ls] : 0;
sum2[root] += sum2[rs] > 0 ? sum2[rs] : 0;
sum1[root] = sum1[ls] + sum1[rs];
}
void build(int root,int l,int r)
{
if(l == r)
{
sum1[root] = sum2[root] = a[l] - a[l-1];
if(l == 1) sum2[root] = 0;
return;
}
int mid = (l + r)>>1;
build(ls,l,mid);
build(rs,mid+1,r);
update(root);
return;
}
void New(int root,int l,int r,int val)
{
if(val > 0) sum2[root] = 1ll * val * (r-l+1);
else sum2[root] = 0;
sum1[root] = 1ll * val * (r-l+1);
det[root] = val;
}
void pushdown(int root,int l,int r)
{
if(det[root] != -INF)
{
int mid = (l + r)>>1;
New(ls,l,mid,det[root]);
New(rs,mid+1,r,det[root]);
det[root] = -INF;
update(root);
}
}
void init()
{
read(n);read(m);
rep(i,1,n) read(a[i]);
build(1,1,n);
return;
}
ll query1(int root,int l,int r,int x,int y)
{
if(l > y || r < x) return 0;
if(l == r) return sum1[root];
if(x <= l && r <= y) return sum1[root];
int mid = (l+r)>>1;
if(det[root] != -INF) pushdown(root,l,r);
ll sum = 0;
sum = query1(ls,l,mid,x,y) + query1(rs,mid+1,r,x,y);
update(root);
return sum;
}
ll query2(int root,int l,int r,int x,int y)
{
if(l > y || r < x) return 0;
if(l == r) return sum2[root] > 0 ? sum2[root] : 0;
if(x <= l && r <= y) return sum2[root];
int mid = (l+r)>>1;;
if(det[root] != -INF) pushdown(root,l,r);
ll sum = 0;
sum = query2(ls,l,mid,x,y) + query2(rs,mid+1,r,x,y);
update(root);
return sum;
}
void change(int root,int l,int r,int x,int y,int val)
{
if(l > y || r < x) return;
if(l == r)
{
if(l != 1)sum2[root] = sum1[root] = val;
else sum1[root] = val;
return;
}
if(x <= l && r <= y)
{
if(val > 0) sum2[root] = 1ll * val * (r-l+1);
else sum2[root] = 0;
sum1[root] = 1ll * val * (r-l+1);
det[root] = val;
return;
}
int mid = (l + r)>>1;
pushdown(root,l,r);
change(ls,l,mid,x,y,val);
change(rs,mid+1,r,x,y,val);
update(root);
}
int main()
{
freopen("gem.in","r",stdin);
freopen("gem.out","w",stdout);
init();
rep(i,1,4*n) det[i] = -INF;
rep(i,1,m)
{
int kind;read(kind);
int l,r;read(l);read(r);
if(kind == 2)
{
int x,y;read(x);read(y);
ll nl = query1(1,1,n,1,l-1);
ll nr = query1(1,1,n,1,r+1);
change(1,1,n,l,l,x-nl);
if(l + 1<=r)change(1,1,n,l+1,r,y);
ll nrr = query1(1,1,n,1,r);
change(1,1,n,r+1,r+1,nr-nrr);
}
else{
printf("%lld\n",query2(1,1,n,l+1,r));
}
}
return 0;
}