第六周 限时大模拟 A - 掌握魔法の东东 II

A - 掌握魔法の东东 II

题目描述

从瑞神家打牌回来后，东东痛定思痛，决定苦练牌技，终成赌神！
东东有 A × B 张扑克牌。每张扑克牌有一个大小(整数，记为a，范围区间是 0 到 A - 1）和一个花色（整数，记为b，范围区间是 0 到 B - 1。
扑克牌是互异的，也就是独一无二的，也就是说没有两张牌大小和花色都相同。
“一手牌”的意思是你手里有5张不同的牌，这 5 张牌没有谁在前谁在后的顺序之分，它们可以形成一个牌型。 我们定义了 9 种牌型，如下是 9 种牌型的规则，我们用“低序号优先”来匹配牌型，即这“一手牌”从上到下满足的第一个牌型规则就是它的“牌型编号”（一个整数，属于1到9）：

同花顺: 同时满足规则 2 和规则 3.
顺子 : 5张牌的大小形如 x, x + 1, x + 2, x + 3, x + 4
同花 : 5张牌都是相同花色的.
炸弹 : 5张牌其中有4张牌的大小相等.
三带二 : 5张牌其中有3张牌的大小相等，且另外2张牌的大小也相等.
两对: 5张牌其中有2张牌的大小相等，且另外3张牌中2张牌的大小相等.
三条: 5张牌其中有3张牌的大小相等.
一对: 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 种牌型的方案数（按牌型编号从小到大的顺序）

Examples
Input
5 2
1 0 3 1
Output
0 8 0 0 0 12 0 36 0
Input
25 4
0 0 24 3
Output
0 0 0 2 18 1656 644 36432 113344

解题思路

（1）一开始想用排列组合把每种牌型分别算出来，发现情况太多，脑力不够用，又看到这题数据比较小，对时间复杂度的要求不高，于是决定暴力枚举。将开局两张牌放入手牌区，剩下的牌单独存放。从剩下的牌中选取三张牌放入手牌区，给牌按牌面和花色排序，然后判断牌型，枚举所有组合，就能得到答案。
（2）判断牌型。只有五张牌，数字关系比较简单，就用枚举来判断牌型了。

代码

#include <iostream>
#include <cmath>
#include <cstdio> 
#include <algorithm>
#define maxn 1000001
using namespace std;

int A,B,m;
long long int ans[10];

struct cd {
	int first,second;
	bool operator< (const cd &p) {
		if(first==p.first)
			return second<p.second;
		else 
			return first<p.first;
	} 
	bool operator== (const cd &p) {
		return first==p.first&&second==p.second;
	}
	cd() {}
	cd(int fir,int sec): first(fir),second(sec) {}
}h[6],t[6],a[maxn];

void judge() {
	//前三种 
	bool flag2=true,flag3=true;
	for(int i=1;i<=4;i++) {
		if(t[i+1].first-t[i].first!=1) flag2=false;  
		if(t[i].second-t[i+1].second) flag3=false;  
	}
	if(flag2&&flag3) { //同花顺 
		ans[1]++; 
		return;
	}
	if(flag2) { //顺子 
		ans[2]++;
		return;
	}
	if(flag3) { //同花 
		ans[3]++;
		return;
	}
	
	//后六种 
	if((t[1].first==t[2].first&&t[1].first==t[3].first&&t[1].first==t[4].first)||
	   (t[5].first==t[2].first&&t[5].first==t[3].first&&t[5].first==t[4].first)) {
	   	//炸弹 
		ans[4]++;	
		return;
	}
	
	if((t[1].first==t[2].first&&t[5].first==t[4].first&&t[5].first==t[3].first)||
	   (t[4].first==t[5].first&&t[1].first==t[2].first&&t[1].first==t[3].first)) {
		//三带二
		ans[5]++;
		return;
	   } 
	if((t[1].first==t[2].first&&t[3].first==t[4].first)||
	   (t[2].first==t[3].first&&t[4].first==t[5].first)||
	   (t[1].first==t[2].first&&t[4].first==t[5].first)) {
	   	//二对
		   ans[6]++; 
		   return;
	   }
	if((t[1].first==t[2].first&&t[1].first==t[3].first)||
	   (t[2].first==t[3].first&&t[2].first==t[4].first)||
	   (t[3].first==t[4].first&&t[3].first==t[5].first)) {
	   	//三条
		   ans[7]++; 
		   return;
	   }
	if(t[1].first!=t[2].first&&t[2].first!=t[3].first&&t[3].first!=t[4].first&&
	t[4].first!=t[5].first) {
		ans[9]++;
		return;
	}
	else {
		ans[8]++;
		return;
	}
}

int main() {
	scanf("%d%d",&A,&B);
	scanf("%d%d%d%d",&h[1].first,&h[1].second,&h[2].first,&h[2].second);
	for(int i=0;i<10;i++) ans[i]=0;
	
	m=0;
	for(int i=0;i<A;i++) { //把剩下的牌挑出来 
		for(int j=0;j<B;j++) {
			if((i!=h[1].first||j!=h[1].second)&&(i!=h[2].first||j!=h[2].second)) {
				m++;
				a[m]={i,j};
			}
		}
	}
	
	for(int i=1;i<=m;i++) { //枚举剩下三张牌 
		for(int j=i+1;j<=m;j++) {
			for(int k=j+1;k<=m;k++) {
				h[3]=a[i],h[4]=a[j],h[5]=a[k];
				for(int i=1;i<=5;i++) {
					t[i]=h[i];
				}
				sort(t+1,t+6); 
				judge();
			}
		}
	}
	for(int i=1;i<=9;i++) {
		printf("%lld",ans[i]);
		if(i!=9) cout<<' ';
	}
	cout<<endl;
	return 0;
}