题目描述
有n根木棍, 第i根木棍的长度为Li,n根木棍依次连结了一起, 总共有n-1个连接处. 现在允许你最多砍断m个连接处, 砍完后n根木棍被分成了很多段,要求满足总长度最大的一段长度最小, 并且输出有多少种砍的方法使得总长度最大的一段长度最小. 并将结果mod 10007。。。
输入格式
输入文件第一行有2个数n,m. 接下来n行每行一个正整数Li,表示第i根木棍的长度.
输出格式
输出有2个数, 第一个数是总长度最大的一段的长度最小值, 第二个数是有多少种砍的方法使得满足条件.
输入输出样例
输入 #1
3 2 1 1 10
输出 #1
10 2
说明/提示
两种砍的方法: (1)(1)(10)和(1 1)(10)
数据范围
n<=50000, 0<=m<=min(n-1,1000).
1<=Li<=1000.
Code:
#include <algorithm>
#include <cstdio>
#define maxn 500010
const int mod = 10007;
inline void reduce(int &x) { x += x >> 31 & mod; }
int n, m, res, ans;
int li[maxn];
int f[2][maxn], now = 1, past = 0;
inline int check(int mid) {
int cnt = 0, res = 0;
for (int i = 1; i <= n; ++i) {
if (cnt + li[i] > mid) {
cnt = 0;
++res;
}
cnt += li[i];
}
return res + static_cast<bool> (cnt);
}
int main() {
scanf("%d%d", &n, &m); ++m;
{
int l = 1, r = 0;
for (int i = 1; i <= n; ++i) {
scanf("%d", li + i);
l = std::max(li[i], l);
r += li[i];
}
while (l <= r) {
int mid = l + r >> 1;
if (check(mid) <= m) r = mid - 1, res = mid;
else l = mid + 1;
}
}
printf("%d ", res);
f[now][0] = 1;
for (int i = 1; i <= m; ++i) {
static const int sz = sizeof f[now];
std::swap(now, past);
__builtin_memset(f[now], 0, sz);
int len = 0, up = f[past][0], lst = 0;
for (int j = 1; j <= n; ++j) {
len += li[j];
while (len > res) {
reduce(up -= f[past][lst]);
len -= li[++lst];
}
f[now][j] = up;
reduce(up += f[past][j] - mod);
}
reduce(ans += f[now][n] - mod);
}
printf("%d\n", ans);
return 0;
}