codeforces B. Cost of the Array

https://codeforces.com/contest/2059/problem/B

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100;
bool cmp(int a,int b)
{
	return a>b;
}
void solve() {
    int n, k;
    cin >> n >> k;
    vector<int> a(n);
    for (auto &it: a) cin >> it;
      cout << a[0]<<endl;
    if ( k == n) {//k==n情况只有一种
        for (int i = 1,j=1; i < n; i += 2,j++) {
            if (a[i] != j) {
                cout << j<< '\n';
                return;
            }
        }
        cout << k/2 + 1 << '\n';
    } else {
//枚举的是第二个子数组的首位是什么，而且是在合理的最大范围内进行枚举，也就是枚举2首可能的值
//最大枚举范围是1~n-（k-1）一共要求是分成k个数组，我们当前枚举的是第二个子数组，保证后面至少有n-2个数，就能保证最终一共有k个子数组
//第1个数一定属于第一个子数组所以从第二位数枚举，如果等于1可以理解成把1给第一个数组，直到找到一个！=1的数
        for (int i = 1; i <= n -k + 1; i++) {
            if (a[i] != 1) {
                cout << "1\n";
                return;
            }
        }
//实在找不到就代表全是1，第一个是1符合，第二个还是1那就不符合了，所以输出2
        cout << "2\n";
    }
}
int main(){
	ios_base::sync_with_stdio(false), cin.tie(nullptr);
	int T;
	cin >> T;
	while(T--) solve();
}