ArcSoft, Inc. is a leading global professional computer photography and computer vision technology company.
There are NN working blocks in ArcSoft company, which form a straight line. The CEO of ArcSoft thinks that every block should have equal number of employees, so he wants to re-arrange the current blocks into KK new blocks by the following two operations:
- merge two neighbor blocks into a new block, and the new block’s size is the sum of two old blocks’.
- split one block into two new blocks, and you can assign the size of each block, but the sum should be equal to the old block.
Now the CEO wants to know the minimum operations to re-arrange current blocks into KK block with equal size, please help him.
Input
First line contains an integer TT, which indicates the number of test cases.
Every test case begins with one line which two integers NN and KK, which is the number of old blocks and new blocks.
The second line contains NN numbers a1a1, a2a2, ⋯⋯, aNaN, indicating the size of current blocks.
Limits
1≤T≤1001≤T≤100
1≤N≤1051≤N≤105
1≤K≤1051≤K≤105
1≤ai≤1051≤ai≤105
Output
For every test case, you should output ‘Case #x: y’, where x indicates the case number and counts from 1 and y is the minimum operations.
If the CEO can’t re-arrange KK new blocks with equal size, y equals -1.
Sample Input
3
1 3
14
3 1
2 3 4
3 6
1 2 3
Sample Output
Case #1: -1
Case #2: 2
Case #3: 3
题意:给你N个数,让你分成K等份。你可以将任意一个数分成两部分,也可以叠加到左右邻近的数。求最少操作数。
那么这个题最关键的地方在于:邻近的数。
理应是签到题,一波递推解决。
#include<iostream>
#include<string.h>
#include<stdio.h>
#include<math.h>
#include <queue>
#include <functional>
using namespace std;
#define LL long long
const LL maxn = 1e5 + 6;
LL a[maxn];
int main()
{
LL T, cas = 0;
scanf("%lld", &T);
while (T--) {
LL n, k; scanf("%lld %lld", &n, &k);
LL sum = 0;
for (LL i = 1; i <= n; i++) {
scanf("%lld", &a[i]);
sum += a[i];
}
if (sum%k != 0) {
printf("Case #%lld: -1\n", ++cas);
continue;
}
LL aim = sum / k;
LL ans = 0;
for (LL i = 1; i<n; i++) {
if (a[i]>aim) {
LL num = a[i] / aim;
ans += num;
if (a[i] % aim == 0) ans--;
else
ans++;
a[i] %= aim;
a[i + 1] += a[i];
}
else if (a[i]<aim) {
ans++;
a[i + 1] += a[i];
}
}
ans += a[n] / aim - 1;
printf("Case #%lld: %lld\n", ++cas, ans);
}
return 0;
}