2020NYIST个人积分赛第五场-I

最新推荐文章于 2020-04-25 00:24:55 发布

Yyj.16

最新推荐文章于 2020-04-25 00:24:55 发布

阅读量206

点赞数

本文链接：https://blog.csdn.net/yyj_16/article/details/105444444

版权

题目

原题：https://vjudge.net/problem/POJ-3273

ZYH学长非常菜，这一天他看到了这一题:

给你一个长度为N的序列，现在需要把他们切割成M个子序列（所以每一份都是连续的），使得每个子序列和均不超过某个值X

但是ZYH学长实在太菜了,这个问题困扰了他很久,你可以帮他敲个代码吗？

输入
多组输入输出
每组数据第一行是2个整数N,M（1<=M<=N<=100000），接着是N行，每行一个整数vj，1<=v[j]<=10000,表示这个序列.

输出
输出X的最小值

输入样例
7 5
100
400
300
100
500
101
400

输出样例
500

思路

简单二分最大值的最小化问题

二分区间为数组里n个值的最大值到数组里n个值的和，查找的是第一个<=mid的值。

l是数组里的最高值，r是数组所有值的和。
遍历整个数组，将子序列的和不超过mid的值归入一个组，统计在当前mid情况下能组成几个组。
如果组数小于m，说明mid偏大，范围变成前一半，反之变为后一半

代码

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <stdlib.h>
#include <string>
const int INF = 0x3f3f3f3f;
using namespace std;
int n, m;
int a[2020202];
bool check(int mid)
{
	int k = 1, cnt = 0, ans = 0, sum = 0; //cnt记录组数 k记录下标
	for (int i = 1; i <= n; i++)
	{
		ans += a[i];
	}
	while (cnt < m && k <= n)
	{
		while (cnt < m && sum + a[k] <= mid && k <= n)
		{
			ans -= a[k];
			sum += a[k];//计算当前组的和
			k++;
		}
		cnt++;//组数+1
		sum = 0; //组数+1后把sum清零 用于计算下一组的和
	}
	if (ans > 0)
		return false;
	else
		return true;
}
int main()
{
	cin >> n >> m;
	int l = 0, r = INF, ans;
	for (int i = 1; i <= n; i++)
		cin >> a[i];
	for (int i = 0; i < 100; i++)
	{
		int mid = (l + r) / 2;
		if (check(mid))
		{
			r = mid;
			ans = mid;
		}
		else
		{
			l = mid;
		}
	}
	cout << ans;
}