CodeForces - 1442B Identify the Operations(思维)

最新推荐文章于 2023-01-19 12:11:08 发布

the best

最新推荐文章于 2023-01-19 12:11:08 发布

阅读量98

点赞数

分类专栏：思维

本文链接：https://blog.csdn.net/weixin_45755679/article/details/116432297

版权

思维专栏收录该内容

46 篇文章 2 订阅

订阅专栏

题意：

给出一个全排列数组 $a$ ,每次选择 $a$ 中的一个数删去，并把它相邻的数的任意一个加入到 $b$ 数组的最后，现在给出这个数组 $a$ 和最后的数组 $b$ ，问有几种删法可以构成数组 $b$

题解：

~~一开始想复杂了，其实没那么难~~

用 $v i s$ 数组标记某个数是否需要加入到数组 $b$ 。然后分类讨论：

$1 .$ 如果要加入的这个数的相邻数都需要加入到数组 $b$ 中,说明无法删除，那么这个数必定不可能被加入到 $b$ 中,那么方案数为0

$2 .$ 如果要加入的这个数的相邻数其中一个需要加入到数组 $b$ 中，那么就肯定是删除另外一个，此时方案数为1

3.如果要加入的这个数的相邻数都不需要加入到数组 $b$ 中，那么任选一个，此时方案数为2

代码：

#pragma GCC diagnostic error "-std=c++11"
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#include<ctime>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int,int> pii;
const int mod=998244353;
const int MAXN=2e5+5;
const int inf=0x3f3f3f3f;
int a[MAXN],b[MAXN];
int p[MAXN],vis[MAXN];
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        int n,k;
        cin>>n>>k;
        for(int i=1;i<=n;i++)
        {
            cin>>a[i];
            p[a[i]]=i;
            vis[a[i]]=0;
        }
        for(int i=1;i<=k;i++)
        {
            cin>>b[i];
            vis[b[i]]=1;
        }
        ll ans=1;
        for(int i=1;i<=k;i++)
        {
            int cnt=0;
            vis[b[i]]=0;
            int pre=p[b[i]]-1;
            int nex=p[b[i]]+1;
            if(pre>=1&&!vis[a[pre]])
            {
                cnt++;
            }
            if(nex<=n&&!vis[a[nex]])
            {
                cnt++;
            }
            ans=ans*cnt%mod;
        }
        printf("%lld\n",ans);
    }
}