2022杭电多校10 Wavy Tree

 input：

3
4
1 7 6 5
6
1 2 3 4 5 6
6
1 1 4 5 1 4

output：

2

4

4

题目大意：给我们一个长度为n的数组，让我们通过一定的操作把这个数组转变成满足对于任意的1<=i<n,ai<min(a[i-1],a[i+1])或者ai>max(a[i-1],a[i+1])，其实就是一种从大小来看的话形成一个如下锯齿的形状：

每次操作可以将任意一个数加一或减一，问最少经过多少次操作可以变成这样的形状。

解题思路：通过我们上面画的图可以看出，这样一种数组无非就是上升下降来回更迭，所以对于一个给定长度的数组来说，我们最后通过操作形成的目标数组无非就两种形状，一种是一开始先上升再下降，另一种是先下降再上升，我们对于这两种情况取一个最小值即可，因为我们取的是最小值，所以当一个数比前一个数大时，我们让它比前一个数大1，小的话就是小1.

上代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N=1e6+10;
long long a[N],b[N];
int n; 
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		{
			scanf("%lld",&a[i]);
			b[i]=a[i];
		}
		if(n<=2)
		{
			printf("0\n");
			continue;
		}
		//计算先上升后下降 
		long long ans1=0,ans2=0;
		int flag1=1,flag2=2;
		for(int i=2;i<n;i++)
		{
			if(flag1==1)
			{
				if(a[i]>a[i-1])
				{
					flag1=2;
				}
				else
				{
					ans1+=abs(a[i-1]-a[i])+1;
					a[i]=a[i-1]+1;
					flag1=2;
				}
			}
			else if(flag1==2)
			{
				if(a[i]<a[i-1])
				{
					flag1=1;
				}
				else
				{
					ans1+=abs(a[i-1]-a[i])+1;
					a[i]=a[i-1]-1;
					flag1=1;
				}
			}
		}
		//因为前面我们能只保证了1~n-1的形状，所以对于最后的末尾要进行特判 
		if(a[n-1]<a[n-2])
		{
			if(a[n-1]>=a[n])
			ans1+=abs(a[n-1]-a[n])+1;
		}
		else if(a[n-1]>a[n-2])
		{
			if(a[n-1]<=a[n])
			ans1+=abs(a[n]-a[n-1])+1; 
		}
		//先下降后上升 
		for(int i=2;i<n;i++)
		{
			if(flag2==1)
			{
				if(b[i]>b[i-1])
				{
					flag2=2;
					continue;
				}
				else
				{
					ans2+=abs(b[i-1]-b[i])+1;
					b[i]=b[i-1]+1;
					flag2=2;
				}
			}
			else if(flag2==2)
			{
				if(b[i]<b[i-1])
				{
					flag2=1;
					continue;
				}
				else
				{
					ans2+=abs(b[i-1]-b[i])+1;
					b[i]=b[i-1]-1;
					flag2=1;
				}
			}
		}
		if(b[n-1]<b[n-2])
		{
			if(b[n-1]>=b[n])
			ans2+=abs(b[n-1]-b[n])+1;
		}
		else if(b[n-1]>b[n-2])
		{
			if(b[n-1]<=b[n])
			ans2+=abs(b[n]-b[n-1])+1; 
		}
		printf("%lld\n",min(ans1,ans2));
	}
	return 0;
}