2021牛客寒假算法基础集训营5-D-石子游戏（差分+思维）

最新推荐文章于 2023-01-17 16:22:48 发布

原创最新推荐文章于 2023-01-17 16:22:48 发布 · 243 阅读

0 ·

CC 4.0 BY-SA版权

简单算法/思维专栏收录该内容

34 篇文章

订阅专栏

题目链接：点击进入

题目：

在这里插入图片描述

题意

有 n 堆石子排成一行，其中第 i 堆石子有 ai 个，可以选择做无数次操作：每次操作把连续相邻的 k 个石子堆中的每堆石子数目加一，问能否让每堆石子的数目都相同

思路

先根据差分数组确定是操作前面的相等数(>0)，还是操作后面的相等数(<0)；
其次根据相等数的个数确定能不能操作，若不能操作或者不能操作整数下，直接输出 -1，否则将操作次数计入答案。

代码

#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define inf 0x3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
//#define int long long
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double eps=1e-7;
const double pai=acos(-1.0);
const int N=2e4+10;
const int maxn=1e6+10;
const int mod=1e9+7;
int t,n,m,k;
ll a[maxn],d[maxn];
int main( )
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
  	cin>>t;
  	while(t--)
  	{
  		cin>>n>>k;
  		for(int i=1;i<=n;i++)
  			cin>>a[i],d[i]=a[i]-a[i-1];
  		ll ans=0,flag=0;
  		for(int i=2;i<=n;i++)
  		{
  			if(d[i]>0)
  			{
  				if((i-1)%k==0)
  					ans+=d[i]*(i-1)/k;
  				else
  				{
  					flag=1;
  					break;
				}
			}
			else if(d[i]<0)
			{
				if(i+k<=n+1)
					d[i+k]+=d[i],ans-=d[i];
				else
				{
					flag=1;
					break;
				}
			}
		}
		if(flag) cout<<-1<<endl;
		else cout<<ans<<endl;
	}
	return 0;
}