Educational Codeforces Round 89 (Rated for Div. 2) E. Two Arrays

题目:click
题意:给定a数组长n,b数组长m,将a数组划分为m段(完全m段全覆盖),第i段的最小值等于b[i],问总共多少种方法。

首先去确定划分的位置,由于第i段可能出现多个b[i],仔细一想其实并不影响,因为划分的时候一段区间内的b[i],最后一个b[i]一定划分在第i段,因为b[i]<b[i+1],正因如此,我们从后往前去找到每段区间的b[i]。之后特判1-b[1]在a中的下标,b[m]-n,其他的区间段符合长度的区间段长度相乘即可(把分界点看做可移动的点)。(在一段区间里找x<b[i+1]的这个点 二分,单调队列都可以,考虑到区间都得扫描直接暴力)(注意细节)
见代码:

#include<cmath>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cstdlib>
#include<istream>
#include<vector>
#include<stack>
#include<set>
#include<map>
#include<algorithm>
#include<queue>
#define inf 0x3f3f3f3f
#define llinf 0x3f3f3f3f3f3f3f3f
#define MAX_len 200005*4
using namespace std;
typedef long long ll;
typedef pair<int,int> PP;
const int mod=998244353;
const int MAXlen=1e5+10;
long double eps=1e-9;
int a[200100],b[200100];
int book[200100];
int main()
{
    int n,m,i,j;
    scanf("%d %d",&n,&m);
    for(i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    for(i=1;i<=m;i++)
    {
        scanf("%d",&b[i]);
    }
    j=m;
    for(i=n;i>=1;i--)
    {
        if(a[i]==b[j])
        {
            book[j]=i;
            j--;
            continue;
        }
    }
    if(j)
    {
        printf("0");
        return 0;
    }
    for(i=1;i<=book[1];i++)
    {
        if(a[i]<b[1])
        {
            printf("0\n");
            return 0;
        }
    }
    for(i=book[m];i<=n;i++)
    {
        if(a[i]<b[m])
        {
            printf("0");
            return 0;
        }
    }
    ll ans=1;
    for(i=1;i<m;i++)
    {
        int temp=-1;
        for(j=book[i+1]-1;j>book[i];j--)
        {
            if(a[j]<b[i+1])
            {
                temp=j;
                break;
            }
        }
        if(temp==-1)
        {
            ans=(ans*(book[i+1]-book[i]))%mod;
            continue;
        }
        for(j=book[i]+1;j<=temp;j++)
        {
            if(a[j]<b[i])
            {
                printf("0\n");
                return 0;
            }
        }
        ans=(ans*ll(book[i+1]-temp))%mod;
    }
    printf("%I64d",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43958964/article/details/106729264