关闭

UVA 714-抄书(二分法)

标签: 算法竞赛入门经典第8章-高效算法设计
45人阅读 评论(0) 收藏 举报
分类:

UVA 714-抄书(二分法)

题意:把一个包含n个正整数(不超过10^7)的序列划分成k个(1<=k<=n<=500)非空的连续子序列,每个正整数恰好是属于一个序列。S(i)表示第i个序列各个数之和,求所有S(i)的最大值尽量小的一种划分。如果有多个解,S(1)尽量小,如果仍有多解,S(2)尽量小,依此类推。
Sample Input
2
9 3
100 200 300 400 500 600 700 800 900
5 4
100 100 100 100 100
Sample Output
100 200 300 400 500 / 600 700 / 800 900
100 / 100 / 100 / 100 100
思路:
枚举一个数X,如果能把序列划分成k个或少于k个连续的子序列,使得所有的S(i)都不超过X,则答案小于或等于X;如果不能,则答案大于X。可以用二分加快枚举。

#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;
typedef long long ll;
const int size = 500 + 5;
int n, k, a[size];
bool is[size];
int is_p(ll num)
{
    ll tot = 0;
    int cur_k = 1;
    for(int i = 0; i < n; i++){
        if(tot + a[i] <= num) tot += a[i];
        else{
            tot = a[i];
            cur_k++;
        }
    }
    return cur_k;
}
void print(ll num)
{
    int rk = k - 1;
    ll tot = 0;
    memset(is, 0, sizeof(is));
    for(int i = n - 1; i >= 0; i--){
        if(tot + a[i] > num || i < rk){
            tot = a[i];
            is[i] = 1;
            rk--;
        }
        else tot += a[i];
    }
    for(int i = 0; i < n-1; i++){
        printf("%d ", a[i]);
        if(is[i]) printf("/ ");
    }
    printf("%d\n", a[n-1]);
}
int main()
{
    int T;
    scanf("%d", &T);
    while(T--){
        scanf("%d%d", &n, &k);
        int maxa = 0;
        ll sum = 0;
        for(int i = 0; i < n; i++){
            scanf("%d", &a[i]);
            maxa = max(maxa, a[i]);
            sum += a[i];
        }
        ll l = (ll)maxa, r = sum;
        while(l < r){
            ll m = (l + r) / 2;
            if(is_p(m) <= k) r = m;
            else l = m + 1;
        }
        print(l);
    }
    return 0;
}

总结:
用二分法把优化问题转化为判定问题

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:65次
    • 积分:20
    • 等级:
    • 排名:千里之外
    • 原创:2篇
    • 转载:0篇
    • 译文:0篇
    • 评论:0条
    文章存档