CodeForces-1442-B. Identify the Operations

原博客：Identify the Operations

题目名称：

Identify the Operations

题目大意：

一个数组a长度为n，一个数组b长度为m，a数组中的元素各不相同，b数组中的元素各不相同。在a数组中要将a${}_i$加入到b${}_i$中，可以将a${}_{i-1}$,a${}_{i+1}$划去，然后ai加入到b数组的右侧。问得到b数组的总方案数，答案对998244353取模。

思路：

Div2的最后一题居然是水题，这是我没想到的，错过了大好的上分时机。要依次将a数组的一些元素加入到b数组中来得到b数组。这里有三种情况：

1）如果a${}_{i-1}$和a${}_{i+1}$都在b数组中。此时他们都还在数组a中，ai想加入b数组中，必须划去a${}_{i-1}$或者a${}_{i+1}$，但是显然无法划去。对于这种情况的选择方案为0.

2）a${}_{i-1}$和a${}_{i+1}$中有一个在b中，而另外一个不在。可以划去那个不在b数组中的。方案数为1.

3）a${}_{i-1}$和a${}_{i+1}$两个数字都不在b中，则有两种方案。

用vis数组来标记ai是否在b数组中。用pos数组表示一个值w在a数组中对应的位置。

代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 200000 + 50;
const int Mod = 998244353;
typedef long long ll;
ll a[maxn], b[maxn],pos[maxn],T,n,m,vis[maxn];
int main(){
    cin >> T;
    while(T--){
        cin >> n >> m;
        for (int i = 1; i <= n;i++){
            cin >> a[i];
            pos[a[i]] = i;
            vis[a[i]] = 0;
        }
        for (int i = 1; i <= m;i++){
            cin >> b[i];
            vis[b[i]] = 1;
        }
        ll ans = 1;
        for (int i = 1; i <= m;i++){
            ll cnt = 0;
            vis[b[i]] = 0;
            if(pos[b[i]]-1>=1&&!vis[a[pos[b[i]]-1]])
                cnt++;
            if(pos[b[i]]+1<=n&&!vis[a[pos[b[i]]+1]])
                cnt++;
            ans = (ans * cnt) % Mod;
        //    cout << cnt << endl;
        }
        cout << ans << endl;
    }
    return 0;
}