ZOJ 4034 Mahjong Sorting（水~）

Description

有$3m$张麻将$C_1,...,C_m,B_1,...,B_m,D_1,...,D_m$和一块白板$W$，在这$3m$张麻将里指定一张幸运麻将，然后从这$3m+1$张麻将里选出$n$张排序，给出排序后的序列，问幸运麻将的可能性有多少种

排序方式为：

1.幸运麻将在最前

2.$C$花色在前，$B$花色在中，$D$花色在后

3.同一花色数字小的在前

4.白板在幸运麻将按$2,3$规则排序的位置

Input

第一行一整数$T$表示用例组数，每组用例首先输入两个整数$n,m$，之后输入$n$张麻将

$(1\le n,m\le 10^5,n\le 3m+1)$

Output

输出幸运麻将的可能方案，保证有解

Sample Input

4
3 9
C 2
W
C 4
6 9
C 2
C 7
W
B 3
B 4
D 2
3 100
C 2
W
C 9
3 9
C 1
B 2
D 3

Sample Output

4
7
25

Solution

给$3m$张牌编号为$1$~$3m$，其中$C_i=i,B_i=m+i,D_i=2m+i$，假设输入的$n$张牌编号为$a[1],...,a[n]$

分几种情况讨论：

$n=1$时，每张麻将都可能是幸运麻将，答案是$3m$

$n>1$时，如果没有白板，如果$a[1]>a[2]$，说明$a[1]$必然是幸运麻将，答案是$1$，否则除了$a[2],...,a[n]$之外的所有牌都有可能，答案是$3m-n+1$

$n=2$时，白板如果在第一个位置，那么幸运麻将编号必然小于$a[2]$，答案为$a[2]-1$；白板如果在第二个位置，那么$a[1]$可能是幸运麻将，如果$a[1]$不是幸运麻将所有大于$a[1]$的都可能是，所以答案是$3m-a[1]+1$

$n>2$时，白板如果在第一个位置，同理答案是$a[2]-1$；白板如果在最后一个位置，那么幸运麻将编号必然大于$a[n-1]$，答案是$3m-a[n-1]$；白板如果在第二个位置，那么幸运麻将只能是$a[1]$~$a[3]-1$，答案是$a[3]-a[1]$；否则假设白板在第$2\le x<n$个位置，如果$a[1]>a[2]$，那么只能$a[1]$是幸运麻将，答案是$1$，否则$a[x-1]+1$~$a[x+1]-1$之间都可以是幸运麻将，答案是$a[x+1]-a[x-1]-1$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100005;
int T,n,m,a[maxn];
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        int pos=0;
        for(int i=1;i<=n;i++)
        {
            char s[3];
            scanf("%s",s);
            if(s[0]=='W')pos=i,a[i]=0;
            else
            {
                if(s[0]=='C')a[i]=0;
                else if(s[0]=='B')a[i]=m;
                else a[i]=2*m;
                int temp;
                scanf("%d",&temp);
                a[i]+=temp;
            }
        }
        if(n==1)printf("%d\n",3*m);
        else if(pos==0)
        {
            if(a[1]<a[2])printf("%d\n",3*m-n+1);
            else printf("1\n");
        }
        else if(n==2)
        {
            if(pos==1)printf("%d\n",a[2]-1);
            else printf("%d\n",3*m-a[1]+1);
        }
        else if(pos==1)printf("%d\n",a[2]-1);
        else if(pos==n)
        {
            if(a[1]>a[2])printf("1\n");
            else printf("%d\n",3*m-a[n-1]);
        }
        else if(pos==2)printf("%d\n",a[3]-a[1]);
        else
        {
            if(a[1]>a[2])printf("1\n");
            else printf("%d\n",a[pos+1]-a[pos-1]-1);
        }
    }
    return 0;
}