题目链接:点击进入
题目:
题意
有 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;
}