题意:
给定长度分别为
n
n
n 和
m
m
m 的两个数组
a
a
a 和
b
b
b ,现在将
a
a
a 划分为
m
m
m 段,要求第
i
i
i 段的最小值等于
b
i
b_i
bi ,求有多少种划分方式;
1
≤
n
,
m
≤
2
⋅
1
0
5
;
1 \leq n,m \leq 2 \cdot 10^5;
1≤n,m≤2⋅105;
1
≤
a
i
≤
1
0
9
;
1 \leq a_i \leq 10^9;
1≤ai≤109;
1
≤
b
i
≤
1
0
9
;
b
i
<
b
i
+
1
1 \leq b_i \leq 10^9;b_i<b_{i+1}
1≤bi≤109;bi<bi+1
分析:
数组
b
b
b 严格递增,所以对于数组
a
a
a 的一种合法划分方式,
b
i
b_i
bi 在
a
a
a 中最后出现的位置必须是第
i
i
i 段,而第
i
i
i 段左边的可调控范围就是往左延申到第一个
<
b
i
< b_i
<bi 的位置,设这个范围区间为
[
l
,
r
]
[l,r]
[l,r] 则第
i
−
1
i-1
i−1 段的右边界和第
i
i
i 段的左边界就等价于在这个区间找一个位置,则这个范围对答案的贡献乘以
(
r
−
l
+
1
)
(r-l+1)
(r−l+1) 倍的;
做一下数组
a
a
a 的后缀最小值,那么第
i
i
i 段的左边界划分区间即
b
i
b_i
bi 在后缀最小值数组中连续出现的位置(因为是后缀最小值数组,所以出现一定连续),那么统计就相对容易了很多;元素值域偏大,用map映射一下
代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define frep(i,a,b) for(int i=a;i>=b;i--)
const int N = 2E5+10;
const ll MOD = 998244353;
int n,m;
ll a[N],b[N],dp[N]; //dp[i]表示前i段的划分方案数
map<ll,ll>mp;
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
cin>>n>>m;
rep(i,1,n)cin>>a[i];
rep(i,1,m)cin>>b[i];
frep(i,n-1,1)a[i]=min(a[i],a[i+1]); //求完后缀最小值后,数组a就变成了非递减序
if(a[1]!=b[1]) cout<<0,exit(0);
rep(i,1,m)mp[b[i]]=i; //映射
dp[1]=1;
rep(i,1,n)if(mp.count(a[i])){
int k=mp[a[i]];
dp[k]=(dp[k]+dp[k-1])%MOD;
/*
第k段的左边界可调控范围即求完后缀最小值数组后,映射后连续的一段k,所以每出现一个
k就代表第k段的左边界(第k-1段的右边界)选择+1,就加一个dp[k-1]
*/
}
cout<<dp[m];
}