LINK
F. Strange Function dp
Examples
inputCopy
11
4 1 3 3 7 8 7 9 10 7 11
3 5 0 -2 5 3 6 7 8 2 4
3
3 7 10
outputCopy
YES
20
inputCopy
6
2 1 5 3 6 5
3 -9 0 16 22 -14
4
2 3 5 6
outputCopy
NO
题意给你一串数组a,每个对应一个p值(删去对应的a所需的消耗,如果是负数那么你就赚了),要求你删去若干个数,使得剩下的数中的递增序列和b是一致的。要你算出最小消耗是?
dp[i]表示达到i这个点之前所需的消耗值最小是
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=200010;
ll a[3*N],b[3*N],p[3*N];
int n,m;
ll dp[3*N];
const ll inf=1e18;
ll c[N*3];//c=cost
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
for(int i=1;i<=n;i++) scanf("%lld",&p[i]);
scanf("%d",&m);
for(int i=1;i<=m;i++)
scanf("%lld",&b[i]);
for(int i=0;i<=m+1;i++) dp[i]=inf ;
for(int i=1;i<=n;i++)
{
int pos=lower_bound(b+1,b+1+m,a[i])-b;
if(pos==m+1) c[m+1]+=p[i];
else if(p[i]<0)
{
c[pos]+=p[i]; //优先考虑删除负值的点
}
}
dp[1]=0ll;
for(int i=1;i<=n;i++)
{
int pos=lower_bound(b+1,b+1+m,a[i])-b;
if(pos==m+1) continue;
if(b[pos]==a[i])
{
ll tp=p[i]>=0ll?0ll:-p[i];//如果这个点的p为负数,又要保留这个点所以要加上这个负数的绝对值和c[pos]里面的进行抵消
dp[pos+1]=min(dp[pos+1],dp[pos]+c[pos]+tp);
}
if(p[i]>0ll)
{
c[pos]+=p[i];//如果选择删去 ,则应该在cost内加上之前没有处理过的正值的p[i]
}
}
if(dp[m+1]>=5e14){//为了避免一些特殊的数据 现在inf值应该设的比之前的要小
printf("NO\n");return 0;
}
printf("YES\n%lld\n",dp[m+1]+c[m+1]);
}