洛谷P2511 [HAOI2008] 木棍分割

题目描述

有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;
}