吉首大学2019年程序设计竞赛.B.干物妹小埋

Posted 2021-03-06 mmasker

题目链接

题解思路：首先对家具按高度排序，离散化后建树，由于小埋只能从后向前走，因此我们按照输入一遍遍更新带权最长上升子序列，Query函数用来求这个最长上升子序列的总权值，更新完以后求一下所有高度中的最大值即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
#define ls l,mid,rt<<1
#define rs mid+1,r,rt<<1|1
#define endl ‘
‘
const int MAXN = 2e5+10;
const double EPS = 1e-12;
map<int,int>mp;

int n;
int a[MAXN],h[MAXN];
ll v[MAXN],mx[MAXN<<2];

void Update(int l,int r,int rt,int pos,ll c){
    if(l==r&&l==pos){
        mx[rt]=c;
        return ;
    }
    int mid=(l+r)/2;
    if(pos<=mid)Update(ls,pos,c);
    else Update(rs,pos,c);
    mx[rt]=max(mx[rt<<1],mx[rt<<1|1]);
}

ll Query(int l,int r,int rt,int L,int R){
    if(L<=l&&R>=r)return mx[rt];
    int mid=(l+r)/2;
    ll ans=0;
    if(mid>=L)ans=max(ans,Query(ls,L,R));
    if(mid<R)ans=max(ans,Query(rs,L,R));
    return ans;
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        h[i]=a[i];
    }
    sort(a+1,a+n+1);
    int cnt=unique(a+1,a+n+1)-a-1;
    for(int i=1;i<=cnt;i++)mp[a[i]]=i;
    for(int i=1;i<=n;i++){
        scanf("%lld",&v[i]);
        ll maxx=Query(1,cnt,1,1,mp[h[i]]);
        Update(1,cnt,1,mp[h[i]],maxx+v[i]);
    }
    cout<<Query(1,cnt,1,1,cnt)<<endl;
}