程序设计思维与实践 Week6 限时大模拟 (3/4/数据班)

从瑞神家打牌回来后，东东痛定思痛，决定苦练牌技，终成赌神！
东东有 A × B 张扑克牌。每张扑克牌有一个大小(整数，记为a，范围区间是 0 到 A - 1）和一个花色（整数，记为b，范围区间是 0 到 B - 1。
扑克牌是互异的，也就是独一无二的，也就是说没有两张牌大小和花色都相同。
“一手牌”的意思是你手里有5张不同的牌，这 5 张牌没有谁在前谁在后的顺序之分，它们可以形成一个牌型。 我们定义了 9 种牌型，如下是 9 种牌型的规则，我们用“低序号优先”来匹配牌型，即这“一手牌”从上到下满足的第一个牌型规则就是它的“牌型编号”（一个整数，属于1到9）：
同花顺: 同时满足规则 5 和规则 4.
炸弹 : 5张牌其中有4张牌的大小相等.
三带二 : 5张牌其中有3张牌的大小相等，且另外2张牌的大小也相等.
同花 : 5张牌都是相同花色的.
顺子 : 5张牌的大小形如 x, x + 1, x + 2, x + 3, x + 4
三条: 5张牌其中有3张牌的大小相等.
两对: 5张牌其中有2张牌的大小相等，且另外3张牌中2张牌的大小相等.
一对: 5张牌其中有2张牌的大小相等.
要不起: 这手牌不满足上述的牌型中任意一个.
现在, 东东从A × B 张扑克牌中拿走了 2 张牌！分别是 (a1, b1) 和 (a2, b2). （其中a表示大小，b表示花色）
现在要从剩下的扑克牌中再随机拿出 3 张！组成一手牌！！
其实东东除了会打代码，他业余还是一个魔法师，现在他要预言他的未来的可能性，即他将拿到的“一手牌”的可能性，我们用一个“牌型编号（一个整数，属于1到9）”来表示这手牌的牌型，那么他的未来有 9 种可能，但每种可能的方案数不一样。
现在，东东的阿戈摩托之眼没了，你需要帮他算一算 9 种牌型中，每种牌型的方案数。
Input
第 1 行包含了整数 A 和 B (5 ≤ A ≤ 25, 1 ≤ B ≤ 4).
第 2 行包含了整数 a1, b1, a2, b2 (0 ≤ a1, a2 ≤ A - 1, 0 ≤ b1, b2 ≤ B - 1, (a1, b1) ≠ (a2, b2)).
Output
输出一行，这行有 9 个整数，每个整数代表了 9 种牌型的方案数（按牌型编号从小到大的顺序）

思路：创建九个条件分别判断抽三张牌后与已经抽的两张牌分别能组成的某种牌型有多少种。对于每个条件，首先判断已经抽的两张牌是否能满足此种牌型，如对于同花顺，当两张牌大小之差大于4或者两者花色不同则不能组成同花顺。对于其他条件也一样。接下来就是对每种牌型进行排列组合看能够得到多少种，特别要注意的是对于同花，它不应该包含同花顺的牌，顺子也如此。

#include <iostream>
#include <algorithm>
using namespace std;
struct pai{
	int dx,hs;
};
int A,B,a[10];
pai take[2];
void yishoupai(int i){
	int ma,mi;
	ma=max(take[0].dx,take[1].dx);
	mi=min(take[0].dx,take[1].dx);
	if(i==1){//同花顺 
		if(abs(take[1].dx-take[0].dx)>4||take[1].hs!=take[0].hs) return;
		if(abs(take[1].dx-take[0].dx)==4) a[1]++;
		else if(abs(take[1].dx-take[0].dx)==3){
			if(ma<A-1) a[1]++;
			if(mi>0) a[1]++;
		}
		else if(abs(take[1].dx-take[0].dx)==2){
			if(ma<A-2) a[1]++;
			if(mi>=2) a[1]++;
			if(ma<A-1&&mi>0) a[1]++;
		}
		else if(abs(take[1].dx-take[0].dx)==1){
			if(ma<A-3) a[1]++;
			if(mi>=3) a[1]++;
			if(ma<A-2&&mi>0) a[1]++;
			if(ma<A-1&&mi>1) a[1]++;
		}
	}
	if(i==2){//炸弹 
		if(B<4) return;
		if(take[0].dx==take[1].dx) a[2]++;
		else if(take[0].dx!=take[1].dx) a[2]+=2;	
	}
	if(i==3){//三带二 
		if(B<3) return;
			if(take[0].dx==take[1].dx){
				int b,c;
				b=B*(B-1)*(B-2)/6;
				a[3]+=(A-1)*b; //作为二
				if(B==4) b=2;
				else b=1;
				c=B*(B-1)/2;
				a[3]+= b*(A-1)*c;//作为三 	
			}
			else if(take[0].dx!=take[1].dx){ //一张为三一张为二 
				int b,c=B-1,num;
				b=(B-1)*(B-2)/2;
				num=2*b*c;
				a[3]+=num; 
			}
	} 
	if(i==4){//同花
		if(take[1].hs!=take[0].hs) return;
		int b=(A-2)*(A-3)*(A-4)/6;
		a[4]+=b-a[1];
	}
	if(i==5){ //顺子 
		if(abs(take[1].dx-take[0].dx)>4) return;
		int b=B*B*B;
		if(abs(take[1].dx-take[0].dx)==4) a[5]+=b;
		else if(abs(take[1].dx-take[0].dx)==3){
			if(ma<A-1) a[5]+=b;
			if(mi>0) a[5]+=b;
		}
		else if(abs(take[1].dx-take[0].dx)==2){
			if(ma<A-2) a[5]+=b;
			if(mi>=2) a[5]+=b;
			if(ma<A-1&&mi>0) a[5]+=b;
		}
		else if(abs(take[1].dx-take[0].dx)==1){
			if(ma<A-3) a[5]+=b;
			if(mi>=3) a[5]+=b;
			if(ma<A-2&&mi>0) a[5]+=b;
			if(ma<A-1&&mi>1) a[5]+=b;
		}
		a[5]-=a[1];
	}
	if(i==6){//三条
		if(B<3) return;
		if(take[1].dx==take[0].dx){
		a[6]=(B-2)*(A-1)*(A-2)*B*B/2;			
	}else{
		a[6]=(A-2)*B*(B-1)*(B-2)/6+(A-2)*B*(B-1)*(B-2);	
	}		
	}
	if(i==7){//两对 
		if(B==1) return;
		if(take[1].dx==take[0].dx){
		a[7]=(A-1)*B*(B-1)*(A-2)*B/2;
			
	}else{
		a[7]=(B-1)*(B-1)*(A-2)*B +(B-1)*(A-2)*B*(B-1); 
	} 
	}
	if(i==8){//一对 
		if(B<2) return;
		if(take[1].dx==take[0].dx){
		a[8]=(A-1)*(A-2)*(A-3)*B*B*B/6; 
		}
		else{
		a[8]=(B-1)*(A-2)*B*(A-3)*B+ (A-2)*B*(B-1)*(A-3)*B/2;	
	} 
	}
	if(i==9){
		int sum=0;
		sum=(A*B-2)*(A*B-3)*(A*B-4)/6; 
		for(int l=1;l<=8;l++) sum-=a[l];
		a[9]=sum;
	}
	
}
int main(){
	cin>>A>>B;
	fill(a,a+10,0);
	for(int i=0;i<2;i++){
		cin>>take[i].dx>>take[i].hs;
	}
	//cout<<take[1].hs<<" "<<take[0].hs<<endl;
	for(int k=1;k<=9;k++) yishoupai(k);
	cout<<a[1];
	for(int i=2;i<=9;i++) cout<<" "<<a[i];
	return 0;
}