noip 1995 灯的排列问题

Description

设在一排上有N个格子（N≤20），若在格子中放置有不同颜色的灯，每种灯的个数记为N1，N2，……Nk（k表示不同颜色灯的个数）。(颜色数<4)
放灯时要遵守下列规则：
①同一种颜色的灯不能分开；
②不同颜色的灯之间至少要有一个空位置。
例如：N=8（格子数）
R=2（红灯数）
B=3（蓝灯数）
放置的方法有： R-B顺序，B-R顺序, 放置的总数为12种。
程序要求：求出排列方案总数。

Input

数据输入的方式为：
N
P1（颜色，为一个字母） N1（灯的数量）
P2 N2
……
Q（结束标记，Q本身不是灯的颜色）

Output

排列方案总数

Sample Input
8
R 2
B 3
Q
Sample Output
12
 

思路：首先可以将同种颜色的灯的个数化为1个，即n-ni+1，灯颜色种数为m

一，DFS,,首先灯之间一定有m-1个空格且每个灯占用一个空格，那么剩下s=n-2*m+1个空格，而灯旁边有m+1个间隙，可以考虑为s个空格插在m+1个间隙中，一个间隙可以插入0，1个或多个空格，那么可以用DFS来遍历间隙即可

二，排列组合：首先灯占用m个空格，那么还剩下n-m个空格，空格旁有n-m+1个间隙，可以考虑为m个灯插在n-m+1个间隙中，每个间隙可以插0个或1个灯，那么所有情况数就为 A(n-m+1,m)

Code 1：

#include<iostream>
using namespace std;

int n,m,ans;

void DFS(int k,int sum);
int main()
{
	ios::sync_with_stdio(false); 
	while(cin>>n){
		ans=m=0;
		char c;
		int x;
		while(cin>>c&&c!='Q'){
			cin>>x;
			n-=x-1;	++m;
		}
		int s=n-2*m+1,p=1;
		for(int i=1;i<=m;++i)
			p*=i;
		m+=1;
		DFS(1,s);
		ans*=p;
		cout<<ans<<endl;
	}
	
	return 0;
}

void DFS(int k,int sum)
{
	if(k==m||!sum){
		++ans;	return;
	}
	for(int i=0;i<=sum;++i)
		DFS(k+1,sum-i);
}

Code 2：

#include<iostream>
using namespace std;

int n;

int main()
{
	ios::sync_with_stdio(false); 
	while(cin>>n){
		char c;
		int x,m=0;
		while(cin>>c&&c!='Q'){
			cin>>x;
			n-=x+1;	++m;
		}
		int ans=1;
		for(int i=2;i<=m+1;++i)
			ans*=(n+i);
		cout<<ans<<endl;
	}
	
	return 0;
}