atcoder DP专项 Q flowers (线段树+DP)

题目大意：给定n束花，每束花有对应的高度和价值。

高度为 h1,h2,h3,h4....,(所有高度都不相同）

价值为a1,a2,a3,a4.....,

求删除一些花，使得 h 满足递增，求剩下花的价值总和最大是多少。

分析：有点像求最长上升子序列，但又有些不同。可以定义 f[i] 为以 h[i] 结尾的最大价值和，则我们需要找到不选第 i 束花的最大价值和，如果直接暴力寻找会超时，因此可以用线段树log(n)来查找。具体解释见代码。

代码：

#include<bits/stdc++.h>

using namespace std;
#define int long long
const int N=2e5+10;

struct node{
    int l,r;
    int v;
}tr[N*4];
int h[N],a[N];

void pushup(int u)
{
    tr[u].v=max(tr[u<<1].v,tr[u<<1|1].v);
}

void build(int u,int l,int r)
{
    if(l==r) tr[u]={l,r};
    else {
        tr[u]={l,r};
        int mid=l+r>>1;
        build(u<<1,l,mid);
        build(u<<1|1,mid+1,r);
    }
}

void modify(int u,int x,int v)
{
    if(tr[u].l==x&&tr[u].r==x) tr[u].v=v;
    else {
        int mid=tr[u].l+tr[u].r>>1;
        if(x<=mid) modify(u<<1,x,v);
        else if(x>mid) modify(u<<1|1,x,v);
        pushup(u);
    }
}

int query(int u,int l,int r)
{
    if(tr[u].l>=l&&tr[u].r<=r) return tr[u].v;
    else {
        int mid=tr[u].l+tr[u].r>>1;
        int ans=0;
        if(l<=mid) ans=max(ans,query(u<<1,l,r));
        if(r>mid) ans=max(ans,query(u<<1|1,l,r));
        return ans;
    }
}
signed main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>h[i];
    for(int i=1;i<=n;i++) cin>>a[i];
    build(1,1,n);
    
    int ans=0;
    for(int i=1;i<=n;i++){//这里查询的是区间[1,h[i]]中的最大总花费
                          //因为题目要求h要上升,因此[1,h[i]]
                          //即为以h[i]结尾的最大总花费
        int t=query(1,1,h[i])+a[i];
        //这里用f[i]可能会更形象,f[i]表示以h[i]结尾的最大总花费
        //cout<<t<<endl;
        ans=max(ans,t);
        modify(1,h[i],t);//更新以h[i]结尾的最大总花费的值
                         //类似于状态转移
    }
    cout<<ans<<endl;
    return 0;
}