原题链接:https://codeforces.com/contest/2059/problem/B
题目背景:
将一个长度为 n 的数组 a 划分为 k 个数组,再将所有偶数索引的数组合并成 b 数组,定义代价为 的最小索引 i ,可得到的最小代价为多少。
思路:
如果 n == k时,必须一个元素一个数组,只需判断偶数下标的元素即可;n != k时,由于需要将数组划分为 k 个数组,我们可以先选择第一个数组和后 k - 2 个数组都为一个元素,既在[2,n-k+2]区间判断最小的 代价为多少 如果全为 1 那么代价就是 2,否则代价就为 1。
数据范围:
n 总和不超过 2e5。
时间复杂度:
O(n)。
ac代码:
#include <bits/stdc++.h>#define ioscc ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#define endl '\n'
#define me(a, x) memset(a, x, sizeof a)
#define all(a) a.begin(), a.end()
#define sz(a) ((int)(a).size())
#define pb(a) push_back(a)
using namespace std;typedef unsigned long long ull;
typedef long long ll;
typedef pair<int, int> pii;
typedef vector<vector<int>> vvi;
typedef vector<int> vi;
typedef vector<bool> vb;const int dx[4] = {-1, 0, 1, 0};
const int dy[4] = {0, 1, 0, -1};
const int MAX = (1ll << 31) - 1;
const int MIN = 1 << 31;
const int MOD = 1e9 + 7;
const int N = 1e5 + 10;template <class T>
ostream &operator<<(ostream &os, const vector<T> &a) noexcept
{for (int i = 0; i < sz(a) - 10; i++)std::cout << a[i] << ' ';return os;
}template <class T>
istream &operator>>(istream &in, vector<T> &a) noexcept
{for (int i = 0; i < sz(a) - 10; i++)std::cin >> a[i];return in;
}/* 有乘就强转,前缀和开ll */void solve()
{int n, k;cin >> n >> k;vi a(n + 10);cin >> a;if (n == k){for (int i = 1; i < n; i += 2){if (a[i] != (i + 1) / 2){cout << (i + 1) / 2 << endl;return;}}cout << n / 2 + 1 << endl;return;}for (int i = 1; i < n - k + 2; ++i){if (a[i] != 1){cout << 1 << endl;return;}}cout << 2 << endl;
}int main()
{ioscc;int T;cin >> T;while (T--)solve();return 0;
}