算法学习日记（四）

题目描述

有 N 堆纸牌，编号分别为 1，2，…, N。每堆上有若干张，但纸牌总数必为 N 的倍数。可以在任一堆上取若于张纸牌，然后移动。
　　移牌规则为：在编号为 1 堆上取的纸牌，只能移到编号为 2 的堆上；在编号为 N 的堆上取的纸牌，只能移到编号为 N-1 的堆上；其他堆上取的纸牌，可以移到相邻左边或右边的堆上。
　　现在要求找出一种移动方法，用最少的移动次数使每堆上纸牌数都一样多。

例如 N=4，4 堆纸牌数分别为：
　　①　9　②　8　③　17　④　6
　　移动3次可达到目的：
　　从 ③ 取 4 张牌放到 ④ （9 8 13 10） -> 从 ③ 取 3 张牌放到 ②（9 11 10 10）-> 从 ② 取 1 张牌放到①（10 10 10 10）。

输入描述 Input Description
第一行N（N 堆纸牌，1 <= N <= 100）
第二行A1 A2 … An （N 堆纸牌，每堆纸牌初始数，l<= Ai <=10000）

输出描述 Output Description
输出至屏幕。格式为：
所有堆均达到相等时的最少移动次数。‘

样例输入 Sample Input
4
9 8 17 6

样例输出 Sample Output
3

数据范围及提示 Data Size & Hint
e

#include<iostream>
#include<stdlib.h>
#include<cmath>
#include<algorithm>
#include<vector>
#include<string.h>
#include<string>
#include<stdio.h>
using namespace std;
int ans;
int fun2(int *a,int n,int num)
{
	int sum=0;
	while(sum<num)
	{
		if(n>0)
		{
		sum+=a[--n]	;	
		}
		else 
		break;
	}
	return n;	
}
void fun(int *a,int avg,int n)
{
if (n==0)
return;
if(n==1)
{
	
	return ;
}
if(a[n-1]-avg>=0)
{
	if(a[n-1]-avg==0)
	{
			fun(a,avg,n-1);		
	}
	else{
	
ans++;
//cout<<"3:"<<ans<<endl;
a[n-2]=a[n-2]+a[n-1]-avg;
fun(a,avg,n-1);
}
}
else
{
	int deep=fun2(a,n-1,avg-a[n-1]);
	int sum=0;
	for(int i=0;i<n-1-deep;i++)
	{
		if(n-1-deep==i+1)
		{
			ans+=i+1;
		//	cout<<"1:"<<ans<<endl;
		a[n-i-2]-=(avg-a[n-1]-sum);
		}
		else
		{ 
		ans+=i+1;
			//cout<<"2:"<<ans<<endl;
		a[n-i-2]=0;
		sum+=a[n-i-2];
		} 
		
		
	}
	fun(a,avg,n-1);	
}
}
int main()
{
int n;
cin>>n;
int *a=new int[n];
for(int i=0;i<n;i++)
{
cin>>a[i];	
} 
int sum=0;
for(int i=0;i<n;i++)
{
	sum+=a[i];
		
}
int avg=sum/n;
fun(a,avg,n);
cout<<ans<<endl;
delete []a;
return 0;
}

//有问题。。。