欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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;
}