提交时间:2023-11-16 16:45:15

运行 ID: 110811

#include<bits/stdc++.h> using namespace std; int n,m,l,r,mid; struct node { int data,next; }sum[100005]; int func(int ll,int rr) { int s=0; for(int x=ll;x!=0;x=sum[x].next) s+=sum[x].data; return s; } int main() { cin>>n>>m; for(int i=1;i<=n;i++) { cin>>sum[i].data; sum[i].next=i+1; } sum[n].next=0; for(int i=0;i<n-m;i++) { l=1,r=n-i,mid=(l+r)/2; while(l<r) { if(func(mid-2,mid-1)<func(mid+1,mid+2)) r=mid-1; else if(func(mid-2,mid-1)>func(mid+1,mid+2)) l=mid+1; mid=(l+r)/2; } sum[l].data+=sum[sum[l].next].data; sum[l].next=sum[sum[l].next].next; } for(int x=1;x!=0;x=sum[x].next) cout<<sum[x].data<<' '; return 0; }