题意:把一个包含m个正整数的序列划分成k个非空的连续子序列。使得所有连续子序列的序列和Si的最大值尽量小。
二分,每次判断一下当前的值是否满足条件,然后修改区间。注意初始区间的范围,L应该为所有正整数中的最大值,否则应该判断时注意。输出解的时候要使字典序最小,所以从后面贪心。
#include<bits/stdc++.h> using namespace std; typedef long long ll; const int maxm = 501; ll p[maxm]; bool vis[maxm]; int k,m; inline bool P(ll x) { int cnt = 1; ll cur = p[0]; for(int i = 1; i < m; i++){ if(cur + p[i]<=x) cur += p[i];//if a[i] > x false else cur = p[i],cnt++; } return cnt<=k; } int main() { //freopen("in.txt","r",stdin); int N; scanf("%d",&N); while(N--){ scanf("%d%d",&m,&k); ll sum = 0,Max = -1;; for(int i = 0; i < m; i++) scanf("%lld",p+i),sum+=p[i],Max = max(Max,p[i]); ll L,R,mid; for(L = Max,R = sum; L < R; P(mid)?R=mid:L=mid+1) mid = (L+R)>>1; memset(vis,0,sizeof(vis)); ll cur = p[m-1]; int kk = k-1; for(int i = m-2; i>=0&&kk; i--) { if(cur+p[i]>L||i<kk) vis[i]=true,cur = p[i],kk--; else cur += p[i]; } printf("%lld",p[0]); if(vis[0]) printf(" /"); for(int i = 1; i < m; i++){ printf(" %lld",p[i]); if(vis[i]) printf(" /"); } putchar(‘\n‘); } return 0; }
时间: 2024-11-09 01:24:07