A - 掌握魔法の东东 II
从瑞神家打牌回来后,东东痛定思痛,决定苦练牌技,终成赌神!
东东有 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 种牌型的方案数(按牌型编号从小到大的顺序)
Examples
Input
5 2
1 0 3 1
Output
0 0 0 0 8 0 12 36 0
Input
25 4
0 0 24 3
Output
0 2 18 0 0 644 1656 36432 113344
做法
已有两张牌,从剩下的牌中选3张,利用dfs计算所有可能的组合;当标记了3张牌时就得到了一种组合,遍历数组a找到被标记的牌,并把组合保存在数组s中,调用plan函数判断这种组合属于哪种类型。
判断过程比较繁琐,bool型c1,c2……对应九种牌型,这些函数的判断方法基本类似于枚举。int型数组c【1~9】对应每种牌型的数量。
代码
#include<iostream>
#include<string.h>
#include<queue>
#include<cmath>
using namespace std;
int c[10];
int mark[200];
int A,B;
int a1,b1,a2,b2;
int n=0;//一共n张牌
int n1,n2;//已有的两张牌下标是n1,n2
struct card
{
int figure,color;
}a[200],s[6];
bool c4()
{
for(int i=1;i<5;i++)
{
if(s[i-1].color!=s[i].color)
return false;
}
return true;
}
bool c5()
{
if(s[0].figure+1==s[1].figure&&s[1].figure+1==s[2].figure&&s[2].figure+1==s[3].figure&&s[3].figure+1==s[4].figure)
return true;
else
return false;
}
bool c1()
{
if(c4()&&c5())
return true;
else
return false;
}
bool c2()
{
if(s[0].figure==s[1].figure&&s[1].figure==s[2].figure&&s[2].figure==s[3].figure)
return true;
else if(s[0].figure==s[1].figure&&s[1].figure==s[2].figure&&s[2].figure==s[4].figure)
return true;
else if(s[0].figure==s[1].figure&&s[1].figure==s[3].figure&&s[3].figure==s[4].figure)
return true;
else if(s[0].figure==s[2].figure&&s[3].figure==s[2].figure&&s[3].figure==s[4].figure)
return true;
else if(s[2].figure==s[1].figure&&s[3].figure==s[2].figure&&s[4].figure==s[3].figure)
return true;
else
return false;
}
bool c3()
{
if(s[0].figure==s[1].figure&&s[1].figure==s[2].figure&&s[3].figure==s[4].figure)
return true;
else if(s[0].figure==s[1].figure&&s[1].figure==s[3].figure&&s[2].figure==s[4].figure)
return true;
else if(s[0].figure==s[1].figure&&s[1].figure==s[4].figure&&s[3].figure==s[2].figure)
return true;
else if(s[0].figure==s[2].figure&&s[3].figure==s[2].figure&&s[1].figure==s[4].figure)
return true;
else if(s[0].figure==s[2].figure&&s[4].figure==s[2].figure&&s[1].figure==s[3].figure)
return true;
else if(s[0].figure==s[3].figure&&s[3].figure==s[4].figure&&s[1].figure==s[2].figure)
return true;
else if(s[1].figure==s[2].figure&&s[3].figure==s[2].figure&&s[0].figure==s[4].figure)
return true;
else if(s[1].figure==s[2].figure&&s[4].figure==s[2].figure&&s[0].figure==s[3].figure)
return true;
else if(s[1].figure==s[3].figure&&s[3].figure==s[4].figure&&s[0].figure==s[2].figure)
return true;
else if(s[3].figure==s[2].figure&&s[3].figure==s[4].figure&&s[1].figure==s[0].figure)
return true;
else
return false;
}
bool c6()
{
if(s[0].figure==s[1].figure&&s[1].figure==s[2].figure)
return true;
else if(s[0].figure==s[1].figure&&s[1].figure==s[3].figure)
return true;
else if(s[0].figure==s[1].figure&&s[1].figure==s[4].figure)
return true;
else if(s[0].figure==s[2].figure&&s[3].figure==s[2].figure)
return true;
else if(s[0].figure==s[2].figure&&s[4].figure==s[2].figure)
return true;
else if(s[0].figure==s[3].figure&&s[3].figure==s[4].figure)
return true;
else if(s[1].figure==s[2].figure&&s[3].figure==s[2].figure)
return true;
else if(s[1].figure==s[2].figure&&s[4].figure==s[2].figure)
return true;
else if(s[1].figure==s[3].figure&&s[3].figure==s[4].figure)
return true;
else if(s[3].figure==s[2].figure&&s[3].figure==s[4].figure)
return true;
else
return false;
}
bool c7()
{
if(s[0].figure==s[1].figure&&(s[2].figure==s[3].figure||s[2].figure==s[4].figure||s[3].figure==s[4].figure))
return true;
else if(s[0].figure==s[2].figure&&(s[1].figure==s[3].figure||s[1].figure==s[4].figure||s[3].figure==s[4].figure))
return true;
else if(s[0].figure==s[3].figure&&(s[2].figure==s[1].figure||s[2].figure==s[4].figure||s[1].figure==s[4].figure))
return true;
else if(s[0].figure==s[4].figure&&(s[2].figure==s[3].figure||s[2].figure==s[1].figure||s[3].figure==s[1].figure))
return true;
else if(s[1].figure==s[2].figure&&(s[0].figure==s[3].figure||s[0].figure==s[4].figure||s[3].figure==s[4].figure))
return true;
else if(s[1].figure==s[3].figure&&(s[2].figure==s[0].figure||s[2].figure==s[4].figure||s[0].figure==s[4].figure))
return true;
else if(s[1].figure==s[4].figure&&(s[2].figure==s[3].figure||s[2].figure==s[0].figure||s[3].figure==s[0].figure))
return true;
else if(s[3].figure==s[2].figure&&(s[0].figure==s[1].figure||s[0].figure==s[4].figure||s[1].figure==s[4].figure))
return true;
else if(s[4].figure==s[2].figure&&(s[0].figure==s[3].figure||s[0].figure==s[1].figure||s[3].figure==s[1].figure))
return true;
else if(s[3].figure==s[4].figure&&(s[2].figure==s[1].figure||s[2].figure==s[0].figure||s[1].figure==s[0].figure))
return true;
else
return false;
}
bool c8()
{
if(s[0].figure==s[1].figure)
return true;
else if(s[0].figure==s[2].figure)
return true;
else if(s[0].figure==s[3].figure)
return true;
else if(s[0].figure==s[4].figure)
return true;
else if(s[1].figure==s[2].figure)
return true;
else if(s[1].figure==s[3].figure)
return true;
else if(s[1].figure==s[4].figure)
return true;
else if(s[3].figure==s[2].figure)
return true;
else if(s[4].figure==s[2].figure)
return true;
else if(s[3].figure==s[4].figure)
return true;
else
return false;
}
void plan()
{
if(c1())
c[1]++;
else if(c2())
c[2]++;
else if(c3())
c[3]++;
else if(c4())
c[4]++;
else if(c5())
c[5]++;
else if(c6())
c[6]++;
else if(c7())
c[7]++;
else if(c8())
c[8]++;
else
c[9]++;
}
//int num=0;
void dfs(int pos,int cnt,int vis[])
{
if(cnt==3)//已经标记了k个数
{
int k=0;
for(int i=0;i<n;i++)
{
if(vis[i]==1)
{
//cout<<a[i].figure<<' '<<a[i].color<<"....";
s[k].color=a[i].color;
s[k].figure=a[i].figure;
k++;
}
}
plan();
//num++;
//cout<<endl;
return;
}
if(pos==n) return; //处理到最后一个数
if(vis[pos]==0)
{
vis[pos]=1;
dfs(pos+1,cnt+1,vis);
if(pos!=n1&&pos!=n2)
vis[pos]=0;//回溯
}
dfs(pos+1,cnt,vis);
}
int main()
{
memset(c,0,10);
memset(mark,0,200);
cin>>A>>B;
cin>>a1>>b1>>a2>>b2;
for(int i=0;i<A;i++)
for(int j=0;j<B;j++)
{
a[n].color=j;
a[n].figure=i;
if(j==b1&&i==a1)
{
mark[n]=1;
n1=n;
}
else if(j==b2&&i==a2)
{
mark[n]=1;
n2=n;
}
n++;
}
dfs(0,0,mark);
for(int i=1;i<10;i++)
cout<<c[i]<<' ';
//cout<<num<<endl;
return 0;
}