洛谷P1181 数列分段Section I

题目描述

对于给定的一个长度为N的正整数数列A[i]，现要将其分成连续的若干段，并且每段和不超过M（可以等于M），问最少能将其分成多少段使得满足要求。

输入输出格式

输入格式：

输入文件divide_a.in的第1行包含两个正整数N，M，表示了数列A[i]的长度与每段和的最大值，第2行包含N个空格隔开的非负整数A[i]，如题目所述。

输出格式：

输出文件divide_a.out仅包含一个正整数，输出最少划分的段数。

输入输出样例

输入样例#1：  复制

5 6
4 2 4 5 1

输出样例#1：  复制

3

说明

对于20%的数据，有N≤10；

对于40%的数据，有N≤1000；

对于100%的数据，有N≤100000，M≤10^9，M大于所有数的最小值，A[i]之和不超过109。

将数列如下划分：

[4][2 4][5 1]

第一段和为4，第2段和为6，第3段和为6均满足和不超过M=6，并可以证明3是最少划分的段数。

题解：

用一个变量tmp进行累加，如果遇到tmp>M就ans++，同时tmp=0；

如果最后做完了tmp不是0，ans++

#include<stdio.h>
#include<algorithm>
#include<iostream>
using namespace std;

int number[100005];
int main()
{
	//freopen("1.txt", "r", stdin);
	int N, M;
	cin >> N >> M;
	int ans = 0;
	int tmp = 0;
	for (int i = 0; i < N; i++)
	{
		cin >> number[i];
		if (tmp+number[i]>=M)
		{
			ans++;
			if (tmp + number[i] == M)
			{
				tmp = 0;
			}
			else
			{

				if (number[i] <= M)
					tmp = number[i];
				else
					tmp = 0;
			}
			//i--;
		}
		else
		{
			tmp += number[i];
		}
	}
	if (tmp != 0) ans++;
	cout << ans;
	return 0; 
}