我的代码：

#include<bits/stdc++.h>
using namespace std;
long long n,k,a[1000001],d[1000001],sx[1000001],sy[1000001],f[1000001];
int main(){
	cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i],d[i]=a[i]-a[i-1];
	for(int i=1;i<=n;i++){
		if(d[i]<=0) sx[i]=sx[i-1]+abs(d[i])+1;
		else sx[i]=sx[i-1];
	}
	cout<<max(abs(sx[k]),sy[k+1]);
	return 0;
}