蒟蒻的线段树小代码交上去 CE,但在本地跑得飞快。
#include<cstdio>
#include<iostream>
#define int long long
using namespace std;
const int N=2e5+10;
int n,h[N],a[N],f[N],ans;
int data[N*4],L[N*4],R[N*4];
void pushup(int p)
{
data[p]=max(data[p*2],data[p*2+1]);
}
void build(int p,int l,int r)
{
L[p]=l,R[p]=r;
if (l==r) return;
int mid=(l+r)/2;
build(p*2,l,mid);
build(p*2+1,mid+1,r);
}
void updata(int p,int x,int v)
{
if (L[p]==R[p])
{
data[p]=v;
return;
}
int mid=(L[p]+R[p])/2;
if (x<=mid) updata(p*2,x,v);
if (mid+1<=x) updata(p*2+1,x,v);
pushup(p);
}
int query(int p,int l,int r)
{
if (l<=L[p]&&R[p]<=r) return data[p];
int mid=(L[p]+R[p])/2,res=0;
if (l<=mid) res=max(res,query(p*2,l,r));
if (mid+1<=r) res=max(res,query(p*2+1,l,r));
return res;
}
signed main()
{
scanf("%lld",&n);
for (int i=1;i<=n;i++) scanf("%lld",h+i);
for (int i=1;i<=n;i++) scanf("%lld",a+i);
build(1,1,n);
for (int i=1;i<=n;i++)
{
f[i]=query(1,1,h[i])+a[i];
ans=max(ans,f[i]);
updata(1,h[i],f[i]);
}
printf("%lld\n",ans);
}
不会真的是用 define int long long 太多遭到报应了吧