noip 1995 灯的排列问题 排列组合 DFS

题目描述

设在一排上有N个格子（N≤20），若在格子中放置有不同颜色的灯，每种灯的个数记为N1，N2，……Nk（k表示不同颜色灯的个数）。
   放灯时要遵守下列规则：
①同一种颜色的灯不能分开；
②不同颜色的灯之间至少要有一个空位置。
 例如：

N=8（格子数）
R=2（红灯数）

B=3（蓝灯数）

   放置的方法有：
      R-B顺序


      B-R顺序

    放置的总数为12种。
    程序要求：求排列总数。

输入格式

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

 颜色和灯的数量之间由一个空格分隔。

输出

输出排列总数。
样例输入
8
R 2
B 3
Q

样例输出
12

题解： 

记录方案的dfs：http://www.cnblogs.com/qscqesze/p/4334157.html

注意同一颜色归到一起
dfs出一种颜色顺序下的所有顺序
再将它乘上N种颜色的排列数N！
即可得到总排列数。。。。

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define mem(a) memset(a,0,sizeof(a))
#define mp(x,y) make_pair(x,y)
const int INF = 0x3f3f3f3f;
const ll INFLL = 0x3f3f3f3f3f3f3f3fLL;
inline ll read(){
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
//
const int maxn = 1e5+10;

int n,s,num,k;
char c;
struct node{
	char x;
	int y;
}q[maxn];

void dfs(int a,int b,int c){   //a表示已用颜色  b表示已放格子数  c表示是否要放空格  
	if(a==num && b==n){
		k++;  //同一颜色顺序下排序数  
		return ;
	}
	if(b > n) return ;
	if(!c) dfs(a+1,b+q[a].y,1);
	dfs(a,b+1,0);
}

int main(){
	n = read();
	while(cin>>c, c!='Q'){
		int b = read();
		s += b;
		bool f = 0;
		for(int i=0; i<num; i++){
			if(q[i].x == c){
				q[i].y += b;
				f = 1;
			}
		}
		if(!f){
			q[num].x = c;
			q[num++].y += b;
		}
	}
	int per=1;
	for(int i=1; i<=num; i++)
		per = per*i;
	if(n - (s+num-1) <= 0) {  //格子不够 
		puts("0");
		return 0;
	}else{
		dfs(0,0,0);
		cout << per*k << endl;
	}

    return 0;
}