codeforces 1366E. Two Arrays

传送门

题意:n个数的数组a,m个数的数组b(递增),将数组a分为m段,使得每段最小值min{a[i]} = b[i],求分割方法数

因为数组b递增,所以预处理出数组a的后缀最小值suf[i],然后对于数组a的每个数a[i],设suf[i]刚好等于b[m](若suf[i]在数组b中找不到,则该点不能作为一个新的段的起始点),则对于第m段a的起点范围是:第一个suf[i]==b[m]和最后一个suf[j]==b[m]之间的数[i,j],终点是后缀suf[i],统计每段的方案数数然后相乘即为结果. 最后需要注意第一段的终点是固定的(最后一个suf[i]=b[1]的数).

#include <bits/stdc++.h>
using namespace std;
typedef long double ld;
typedef long long ll;

const int mod = 998244353;


int mult(int a, int b) {
    return (1LL * a * b) % mod;
}


int n, m;
const int maxN = 2e5 + 10;
int a[maxN];
int b[maxN];
int suf[maxN];
int cnt[maxN];
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
    }
    for (int i = 1; i <= m; i++) {
        cin >> b[i];
    }
    suf[n + 1] = 1e9 + 10;
    for (int i = n; i >= 1; i--) {
        suf[i] = min(suf[i + 1], a[i]);
    }
    if (suf[1] != b[1]) {
        cout << 0;
        return 0;
    }
    for (int i = 1; i <= n; i++) {
        int p = suf[i];
        int pos = lower_bound(b + 1, b + m + 1, p) - b;
        if (1 <= pos && pos <= m && b[pos] == p) cnt[pos]++;
    }
    int ans = 1;
	cout<<endl;
    for (int i = 2; i <= m; i++) {
        ans = mult(ans, cnt[i]);
    }
    cout << ans;
    return 0;
}