CS 300 Bounded Difference 模拟

CS 300
题意:给出一个序列a.操作:交换序列中任意两个数.
n<=1e5,a[i],k<=1e9. 问最多操作一次的情况下,能否满足|a[i]-a[i+1]|<=k [i=1..n-1].?

找到第一个|a[i]-a[i+1]|>k的位置 现在要么把i或者i+1交换出去.
假如a[i+1]和某个位置p交换,则需要满足 |a[i+1]-a[p-1]|<=k,|a[i+1]-a[p+1]|<=k && |a[i]-a[p]|<=k.
如果有多个p 如何选择? 随便选一个 水过了??

正解:最多只能有4个错误pair,标记每个pair位置i,

与i交换的位置j只能为(错误pair中的某个位置才会有解).交换(i,j),判定相邻是否合法即可 O(4*N).

#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define ran 111111
int n,dif;
int a[ran],pos[ran],lpos;
bool bad[ran];
bool valid(int x){
	return -dif <= x && x <= dif;
}
int main(){
	scanf("%d%d",&n,&dif);
	for(int i=1; i<=n; i++){
		scanf("%d",&a[i]);
		if(i>1 && !valid(a[i] - a[i-1])){
			pos[lpos ++] = i;
			bad[i] = true;
		}
	}
	if(lpos > 4){
		cout << -1 <<endl;
		return 0;
	}
	if(lpos == 0){
		cout << 0 <<endl;
		return 0;
	}
	for(int i=1; i<=n; i++)
        if(bad[i] || bad[i+1])
	for(int j=1; j<=n; j++)if(i != j){
		bool flag = true;
		for(int k=0; k<lpos; k++)
			if(!(i == pos[k] || i == pos[k] - 1 ||
			j == pos[k] || j == pos[k] - 1)){
				flag = false;
				break;
			}
		if(!flag)continue;
		swap(a[i], a[j]);
		if(
		(i==1 || valid(a[i]-a[i-1])) &&
		(i==n || valid(a[i+1]-a[i])) &&
		(j==1 || valid(a[j]-a[j-1])) &&
		(j==n || valid(a[j+1]-a[j]))
		){
			if(i > j)swap(i,j);
			cout << i << " " << j << endl;
			return 0;
		}
		swap(a[i], a[j]);
	}
	cout << -1 << endl;
	return 0;
}