pat Data Stucture 海盗分赃

#include<iostream>
#include<algorithm>

using namespace std;
int Median(int a[],int start,int end)
{
sort(a+start,a+end);
return a[(end-start-1)/2+start];
}
int main()
{
int d,p;
cin>>d>>p;
int a[p],b[p];
a[p-2]=0;
a[p-1]=d;
for(int i=0;i<p-2;++i)
{
for(int j=p-2-i;j<p;++j)
{
b[j]=a[j];
}
int mid=Median(b,p-2-i,p);
int count=0;
for(int j=p-2-i;j<p;++j)
{
if(a[j]<=mid) {a[j]+=1;++count;}
else a[j]=0;
}
int j=p-1;
while(count>(i+3)/2)
{
if(a[j]-1==mid)
{
a[j]=0;
--j;
--count;
}
else --j;
}
int sum=0;
for(int j=p-2-i;j<p;++j)
{
sum+=a[j];
}
a[p-3-i]=d-sum;
}
cout<<a[0];
}