2021牛客寒假算法基础集训营5 D

石子游戏

开始补题！！！

dif 差分数组

差分的题目，直接预处理出石子的查分数组，若操作后差分值全为0就输出操作次数就好。

题目固定了每次增加的区间大小，所以只需要从左往右边扫一遍，把差分数组从2开始所有的负数（代表前面值比当前值大）变成0（对当前i 到i+k的区间进行增加操作使其为0 dif[i+k]+=dif[i] ),因为此时不让当前位置增加到0，扫描就过去了，无法再使当前位置为0。操作次数ans+=dif[i] (当前位置需要增加的石子数)。

当然这样的操作只是把负数变0，所以还需要从后往前遍历，把正的变0，思想是一样的。

为什么要扫负数的时候要从前往后遍历，而扫正数从后往前遍历，因为解题突破口找到一个位置，如果不进行操作就失去了将它变0的机会，借此可以定位出必须增加的区间和增加的次数。

操作完成后再从2遍历，此时差分数组如果全部都相等那 应该有dif[2]=dif[3]=…dif[n]。所以如果找到

dif[i]不为0，就表明无法做到使该数组为0了，因为在前面的操作中已经在尽量的让所有数相等，相当于你已经做到了这样一组数据 5个数字 k为3 样例： 3 3 1 3 3 这个时候显然是没有办法实现的

AC代码

#include"bits/stdc++.h"
using namespace std;
long long a[100010],dif[100010];
int main()
{
	int t,n,ma,js,k;
    long long ans;
	scanf("%d",&t);
	while(t--)
	{
		ma=0;
		ans=0;
		js=0;
		scanf("%d%d",&n,&k);
		for(int i=1;i<=n;i++)
		{
			scanf("%lld",&a[i]);
			dif[i]=a[i]-a[i-1];
		}
		for(int i=n;i>=k+1;i--)
			if(dif[i]>0)
			{
				dif[i-k]+=dif[i];
                ans+=dif[i];
				dif[i]=0;
			}
		for(int i=2;i<=n-k+1;i++)
			if(dif[i]<0)
			{
				dif[i+k]+=dif[i];
                ans-=dif[i];
				dif[i]=0;
			}
		for(int i=2;i<=n;i++)
		if(dif[i])
		{
			ans=-1;
			break;
		}
		printf("%lld\n",ans);
	}
	return 0;
}