【Week 6 限时大模拟】A - 掌握魔法の东东 II

程序员文章站 2022-05-08 22:13:27

...

A - 掌握魔法の东东 II

题意：

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

1.同花顺: 同时满足规则 2 和规则 3.
2.顺子 : 5张牌的大小形如 x, x + 1, x + 2, x + 3, x + 4
3.同花 : 5张牌都是相同花色的.
4.炸弹 : 5张牌其中有4张牌的大小相等.
5.三带二 : 5张牌其中有3张牌的大小相等，且另外2张牌的大小也相等.
6.两对: 5张牌其中有2张牌的大小相等，且另外3张牌中2张牌的大小相等.
7.三条: 5张牌其中有3张牌的大小相等.
8.一对: 5张牌其中有2张牌的大小相等.
9.要不起: 这手牌不满足上述的牌型中任意一个.

现在, 东东从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 种牌型的方案数（按牌型编号从小到大的顺序）

Sample Input

Sample Output

0 8 0 0 0 12 0 36 0

0 0 0 2 18 1656 644 36432 113344

思路做法：

代码分为2部分，backtrack()递归回溯找出所有的一手牌和check()判断牌型。backtrack()递归向数组pai中加入牌，当牌数到5时，将牌复制给数组p，并调用check()，判断数组p中的5张牌属于哪种牌型并更新sign数组。

总结:

check()时不能用原来的pai数组，sort会打乱顺序。

代码:

#include <stdio.h>
#include <map>
#include <algorithm>
using namespace std;

const int MAXA = 26, MAXB = 4;
int A, B, c, sign[10];
struct Pai{
	int a, b;
	Pai(){}
	Pai(int _a, int _b):a(_a), b(_b){}
	bool operator<(const Pai& pai) const {
		if(a != pai.a) return a < pai.a;
		return b < pai.b;
	}
	void output(){ printf("%d %d\n", a, b); }
}p[6], pai[6];
map<Pai, bool> isUse;

void check(){
	bool two = 1, three = 1;
	map<int, int> cnt;
	sort(pai+1, pai+6);
	for(int i = 1; i <= 5; ++i){
		cnt[pai[i].a]++;
		if(i >= 2){
			if(pai[i].a != pai[i-1].a + 1) two = 0;
			if(pai[i].b != pai[i-1].b) three = 0;
		}
	}
	if(two && three){ sign[1]++; return; }
	if(two){ sign[2]++; return; }
	if(three){ sign[3]++; return; }
	map<int, int>::iterator iter;
	iter = cnt.begin();
	int num[5], n = 0;
	while(iter != cnt.end()){
		num[n++] = iter->second;
		iter++;
	}
	sort(num, num+n, greater<int>());
	if(num[0] == 4) { sign[4]++; return; }
	else if(n == 2 && num[0] == 3 && num[1] == 2) { sign[5]++; return; }
	else if(n == 3 && num[0] == 2 && num[1] == 2) { sign[6]++; return; }
	else if(n == 3 && num[0] == 3 && num[1] == 1) { sign[7]++; return; }
	else if(n == 4 && num[0] == 2) { sign[8]++; return; }
	else { sign[9]++; return; }
}

void backtrack(int a, int b, int s){
	if(s >= 5){
		for(int i = 1; i <= 5; ++i){
			pai[i] = p[i];
		}
		check();
		return;
	}
	for(int k = a*B+b+1; k < A*B; ++k){
		int i = k / B, j = k % B;
		if(!isUse[Pai(i, j)]){
			p[s+1] = Pai(i, j);
			isUse[p[s+1]] = 1;
			backtrack(i, j, s+1);
			isUse[p[s+1]] = 0;
		}
	}
}

int main(){
	scanf("%d %d", &A, &B);
	scanf("%d %d %d %d", &p[1].a, &p[1].b, &p[2].a, &p[2].b);
	for(int i = 1; i <= 9; ++i) sign[i] = 0;
	isUse[p[1]] = 1; isUse[p[2]] = 1;
	c = 0; backtrack(0, -1, 2);
	for(int i = 1; i <= 9; ++i) printf("%d ", sign[i]);
	printf("\n");
	return 0;
}