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];
}