代码里有详细解释
#include <cstdio>
#include <cstring>
#include <climits>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long LL;
#define pEnter putchar('\n')
#define sc(n) scanf("%d",&(n))
#define pInt(n) printf("%d\n", (n))
#define plInt(n) printf("%lld\n", (n))
#define scc(n, m) scanf("%d%d",&(n), &(m))
#define sccc(n, m, k) scanf("%d%d%d",&(n), &(m), &(k))
#define mem(arr, val) memset(arr, val, sizeof(arr))
#define pVarInt(v) printf("%s = %d\n",#v,(v))
#define rep(var, s, t) for(int var=(s); var<=(t); var++)
#define drep(var, s, t) for(int var=(s); var>=(t); var--)
inline void PrintArrInt(int arr[], int s, int e){rep(i, s, e)printf("%d%c",arr[i],i==e?'\n':' ');}
const int maxn = 5e5 + 5;
const int INF = 0x3f3f3f3f;
int N, K, a[maxn], b[maxn], t[maxn], cnt;
/*CF 1437E
题意:给定一个数组,其中有些位置的数字固定不能改,其他位置可以随便改值,问使整个序列严格上升的最小改动次数(若无法达成输出-1)
思路:首先先不管约束,即对于一个序列,改最少次,让它变成递增序列
将所有数字减去它的索引位置,即 a[i]-=i
只要找到处理后的数组的 最长 不下降 子序列的长度为len ,这些值就不用改,其他值要改,故答案为N-len
这里用到了一个转化,因为如果题目不要求严格递增的话,其实对原数组求一个 最长 不下降 子序列就可以了
但是由于要严格递增,例如 3 6 4 5,这样的就不行了,6改成3或4都不满足,即不能保证可以有合法的修改方法
a[i]-=i后,很明显,相同的值就可以满足条件,例如 3 3 3,在原来的数组中为 4 5 6
求得最长 不下降 子序列后,总有方法把其他值改好:3 6 4 5->3 4 4 5
现在加上限制,对于固定的两个位置之间,相当于求必须包含左右端点的 最长 不下降 子序列
先将左端点加入序列,在贪心过程中保证不换掉左端点,完成后,在序列中找到右端点即可得到一定包含端点的序列
*/
int solve(int L, int R) {
cnt = 0; t[++cnt]=a[L];
rep(i, L+1, R) {//求最长 不下降 子序列
if(a[i]>=t[cnt]) t[++cnt]=a[i];
else {
int p=upper_bound(t+1, t+cnt+1, a[i])-t;//!!!!由于是不下降,值可以相同,所以在贪心时要把>a[i]的值"拉"下来,如果用lower_bound对于
if(p!=1) t[p]=a[i]; //这样的序列就求错了:3 4 5 3 3 3 3 3 4
}
}
int p=upper_bound(t+1, t+cnt+1, a[R])-t-1;//!!!!由于a[R]最后一个加入序列,所以一定在与a[R]值相同的一段序列的最右端,必须用upper_bound往
return R-L+1-p; //最右边找
}
int main() {
scc(N, K);
rep(i, 1, N) sc(a[i]), a[i]-=i;
rep(i, 1, K) sc(b[i]);
bool isok = true;
b[0]=0, b[K+1]=N+1;
a[0]=-INF, a[N+1]=INF;
rep(i, 1, K) if(a[b[i]]>a[b[i+1]]) {isok=false; break;}
if(!isok) pInt(-1);
else {
int ans = 0;
rep(i, 0, K) {
int L=b[i], R=b[i+1];
ans+=solve(L, R);
}
pInt(ans);
}
return 0;
}