游游的k-好数组

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网

题目描述

游游定义一个数组为k-好数组，当且仅当它的所有长度为k的区间和相等。例如，数组[1,2,2,1]为3-好数组，因为两个长度为3的区间和都是5。
游游拿到了一个大小为n的数组，每次操作可以选择一个元素加1。游游最多可以进行xxx次操作，她希望不超过x次操作后，数组变为k-好数组，且数组的最大值尽可能大。你可以帮游游求出这个最大值吗？

输入描述

输出描述

示例

输入

3
4 3 1
1 2 1 1
4 2 5
1 2 3 4
5 3 1
1 100 1 1 1

输出

3
4
-1  

思路

例如[1 2 2 1]数组，长度为3的数组为[1 2 2] 和 [2 2 1],有此例子可以知道，两个区间必有2个元素是一样的，要使所有长度为3的区间和相等，那么两个区间里面的元素必定要相同，即第i个元素的值要等于第i+k个元素的值，才是k-好数组。

具体思路见代码注释，建议画个图模拟一遍，这个思路挺巧妙的，我也是学习别人的。

 

代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e5+10;
int a[N];
int n,k,x;
vector<int> tr[N];  //用来存i%k的值，即i,i+k……等的值
void solve(){
    cin>>n>>k>>x;    
    for(int i=0;i<k;i++) tr[i].clear();  //初始化
	for(int i=0;i<n;i++){
		cin>>a[i];
        tr[i%k].push_back(a[i]);  //每一组存的是a[i],a[i+k]……等等
	}
    int cnt=0;
    for(int i=0;i<k;i++){
        sort(tr[i].begin(),tr[i].end()); //排序
        for(int v:tr[i]){         //全部变成最大的那个值需要操作的次数
            cnt+=tr[i].back()-v;  //因为每次操作只能加，所以最终每组的数都会变成最大的那个数
        }
    }
    x-=cnt;    //总操作减去上面操作的次数
    if(x<0){   //小于0则无法变成好数组
        cout<<-1<<'\n';
        return ;
    }
    int ans=0;
    for(int i=0;i<k;i++){  //找剩余操作数操作完后的最大数
        ans=max(ans,tr[i].back());
        int len=tr[i].size();
        ans=max(ans,(int)tr[i].back()+x/len); //x/len是因为每个数都要同时加
    }
    cout<<ans<<'\n';
}
signed main(){
    int t;
    cin>>t;
    while(t--) solve();
}