洛谷 P1181,1182 数列分段Section
题目描述
对于给定的一个长度为N的正整数数列A[i],现要将其分成连续的若干段,并且每段和不超过M(可以等于M),问最少能将其分成多少段使得满足要求。
输入输出格式
输入格式:
输入文件divide_a.in的第1行包含两个正整数N,M,表示了数列A[i]的长度与每段和的最大值,第2行包含N个空格隔开的非负整数A[i],如题目所述。
输出格式:
输出文件divide_a.out仅包含一个正整数,输出最少划分的段数。
输入输出样例
5 6 4 2 4 5 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是最少划分的段数。
直接贪心即可,从左往右加,大于M则另起一段,最后段数即是答案。
1 #include<cstdio> 2 #include<cstdlib> 3 #include<cstring> 4 #include<iostream> 5 #include<cmath> 6 #include<algorithm> 7 #include<vector> 8 #include<stack> 9 #include<queue> 10 #include<map> 11 #define RG register 12 #define IL inline 13 #define pi acos(-1.0) 14 #define ll long long 15 using namespace std; 16 int n,m,a[100005]; 17 int main() { 18 scanf("%d%d",&n,&m); 19 for(int i=1;i<=n;i++) 20 scanf("%d",&a[i]); 21 int ans=0; 22 for(int i=1;i<=n;i++){ 23 int tot=0; 24 while(tot+a[i]<=m&&i!=n+1) tot+=a[i],++i; 25 --i; 26 ++ans; 27 } 28 cout<<ans; 29 return 0; 30 }
题目描述
对于给定的一个长度为N的正整数数列A[i],现要将其分成M(M≤N)段,并要求每段连续,且每段和的最大值最小。
关于最大值最小:
例如一数列4 2 4 5 1要分成3段
将其如下分段:
[4 2][4 5][1]
第一段和为6,第2段和为9,第3段和为1,和最大值为9。
将其如下分段:
[4][2 4][5 1]
第一段和为4,第2段和为6,第3段和为6,和最大值为6。
并且无论如何分段,最大值不会小于6。
所以可以得到要将数列4 2 4 5 1要分成3段,每段和的最大值最小为6。
输入输出格式
输入格式:
输入文件divide_b.in的第1行包含两个正整数N,M,第2行包含N个空格隔开的非负整数A[i],含义如题目所述。
输出格式:
输出文件divide_b.out仅包含一个正整数,即每段和最大值最小为多少。
输入输出样例
5 3 4 2 4 5 1
6
说明
对于20%的数据,有N≤10;
对于40%的数据,有N≤1000;
对于100%的数据,有N≤100000,M≤N, A[i]之和不超过10^9。
要求最大值最小,考虑二分答案。check时与数列分段Section I类似,若段数大于M,不合法,否则合法,继续二分即可。需要注意的是可能存在一个元素直接大于二分的答案的情况,这时需要直接返回不合法,否则i会一直自减,陷入死循环。
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<cstdlib> 5 using namespace std; 6 int v[100005],n,m,ans; 7 bool check(int x){ 8 int ans=0; 9 for(int i=1;i<=n;i++){ 10 int tot=0; 11 while(tot+v[i]<=x&&i<=n) tot+=v[i],++i; 12 if(v[i]>x) return false; 13 ++ans; 14 if(i>n||i==0) break; 15 --i; 16 } 17 if(ans>m) return false; 18 else return true; 19 } 20 int main(){ 21 scanf("%d%d",&n,&m); 22 for(int i=1;i<=n;i++) scanf("%d",&v[i]); 23 int l=1,r=1000000000; 24 while(l<=r){ 25 int mid=(l+r)/2; 26 if(check(mid)) ans=mid,r=mid-1; 27 else l=mid+1; 28 } 29 cout<<ans; 30 return 0; 31 }