算法训练 未名湖边的烦恼【c++】

问题描述

　　每年冬天，北大未名湖上都是滑冰的好地方。北大体育组准备了许多冰鞋，可是人太多了，每天下午收工后，常常一双冰鞋都不剩。
　　每天早上，租鞋窗口都会排起长龙，假设有还鞋的m个，有需要租鞋的n个。现在的问题是，这些人有多少种排法，可以避免出现体育组没有冰鞋可租的尴尬场面。（两个同样需求的人（比如都是租鞋或都是还鞋）交换位置是同一种排法）

输入格式

　　两个整数，表示m和n

输出格式

　　一个整数，表示队伍的排法的方案数。

样例输入

3 2

样例输出

5

数据规模和约定

　　m,n∈［0,18］

/************************************************
*解题思路：
*1.数据规模不大，采用全排列模拟法，可以next_permutation（）
*2.当前面还鞋的人数大于等于借鞋人数，则为成功方案
*3.eg：成功方案：11001，11100，10101...
*      失败方案：00111，01110，01010...
*************************************************/
#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
	int m,n;
	cin >> m >> n;
	int a[100] = {0};  //模拟排队队伍，0为借鞋的，1为还鞋的
	if(m < n)
	{
		cout << 0;
		return 0;
	}
	for(int i = m+n-1;i > n-1;i --)
	{
		a[i] = 1;
	}
	int la = 0,lb = 0;
	int flag;
	int count = 0;
	 
	do
	{
		la = 0;  //保存队列前面还鞋人数
		lb = 0;  //保存队列前面借鞋人数
		flag = 0;
		
		for(int i = 0;i < m+n;i++)
		{
			if(la < lb)   //当队伍前面借鞋的人数比已经换鞋的数量多，则失败，遍历下一个方案
			{ 
				flag = 1;
				break;
			}
			if(a[i] == 1)
			{
				la++;
			}
			else
			{
				lb++;
			}
		}
		if(flag == 0)
		{
			count++;
		}
	}while(next_permutation(a,a+n+m));
	cout << count ;
}