【洛谷 P1181】数列分段 Section I 题解(贪心算法)

数列分段 Section I

题目描述

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

输入格式

第1行包含两个正整数 N , M N,M N,M,表示了数列 A i A_i Ai 的长度与每段和的最大值,第 2 2 2 行包含 N N N 个空格隔开的非负整数 A i A_i Ai,如题目所述。

输出格式

一个正整数,输出最少划分的段数。

样例 #1

样例输入 #1

复制代码
5 6
4 2 4 5 1

样例输出 #1

复制代码
3

提示

对于 20 % 20\% 20%的数据,有 N ≤ 10 N≤10 N≤10;

对于 40 % 40\% 40%的数据,有 N ≤ 1000 N≤1000 N≤1000;

对于 100 % 100\% 100%的数据,有 N ≤ 100000 , M ≤ 1 0 9 N≤100000,M≤10^9 N≤100000,M≤109, M M M大于所有数的最大值, A i A_i Ai之和不超过 1 0 9 10^9 109。

将数列如下划分:

4 \] \[ 24 \] \[ 51 \] \[4\]\[2 4\]\[5 1\] \[4\]\[24\]\[51

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


思路

从前往后遍历数列,每次将当前数加入已有的一段中,如果该段的和超过了 m,则将当前数新开一段。

定义两个变量 cnt 和 sum,分别表示当前已有的分段数和当前分段的和。初始时,cnt 为 1(最后一段),sum 为 0。遍历数列,对于每个数,如果将其加入当前分段后,分段和不超过 m,则将其加入当前分段中,更新 sum 的值。否则,将当前数新开一段,更新 cnt 和 sum 的值。最后输出 cnt 即可。


AC代码

cpp 复制代码
#include <iostream>
#define AUTHOR "HEX9CF"
using namespace std;

const int N = 1e5 + 5;

int main()
{
    int n, m;
    int a[N];
    int cnt, sum;
    cin >> n >> m;
    cnt = 1; // 最后一段
    sum = 0;
    for (int i = 0; i < n; i++)
    {
        cin >> a[i];
        if (sum + a[i] > m)
        {
            sum = a[i];
            cnt++;
        }
        else
        {
            sum += a[i];
        }
    }
    cout << cnt << endl;
    return 0;
}
相关推荐
岁忧43 分钟前
(LeetCode 面试经典 150 题 ) 11. 盛最多水的容器 (贪心+双指针)
java·c++·算法·leetcode·面试·go
chao_7891 小时前
二分查找篇——搜索旋转排序数组【LeetCode】两次二分查找
开发语言·数据结构·python·算法·leetcode
秋说3 小时前
【PTA数据结构 | C语言版】一元多项式求导
c语言·数据结构·算法
Maybyy3 小时前
力扣61.旋转链表
算法·leetcode·链表
卡卡卡卡罗特5 小时前
每日mysql
数据结构·算法
chao_7895 小时前
二分查找篇——搜索旋转排序数组【LeetCode】一次二分查找
数据结构·python·算法·leetcode·二分查找
蜉蝣之翼❉6 小时前
CRT 不同会导致 fopen 地址不同
c++·mfc
aramae6 小时前
C++ -- STL -- vector
开发语言·c++·笔记·后端·visual studio
lifallen6 小时前
Paimon 原子提交实现
java·大数据·数据结构·数据库·后端·算法
lixzest6 小时前
C++ Lambda 表达式详解
服务器·开发语言·c++·算法