(2020牛客暑期多校训练营)[二]Greater and Greater

题目描述
给定大小为n的序列A和大小为m的序列B,计算A中所有大小为m的子区间S,满足:
在这里插入图片描述
样例
输入

6 3
1 4 2 8 5 7
2 3 3

输出

2

思路
看看官方题解和这篇博客园的分析
(2020牛客暑期多校训练营)[二]Greater and Greater_第1张图片
本质上这道题是动态规划,只是用了bitset优化。
代码

#include
using namespace std;
const int N=150010,M=40010;
bitset<M> a,b[N];
int n,m,ans,c[N],d[N],e[N];
int cmp(int x,int y){return d[x]<d[y];}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++) scanf("%d",&c[i]);
    for(int i=0;i<m;i++) scanf("%d",&d[i]),e[i]=i;
    sort(e,e+m,cmp);
    for(int i=1;i<=m;i++) b[i]=b[i-1],b[i].set(e[i-1]);//求出Si
    for (int i=n-1;i>=0;i--)
    {
        a>>=1;int l=0,r=m;
        while(l<r)//二分查找Si
        {
            int mid=l+r>>1;
            if(c[i]<d[e[mid]]) r=mid;
            else l=mid+1;
        }
        a&=b[l];
        if(c[i]>=d[m-1]) a.set(m-1);
        ans+=a[0];
    }
    cout<<ans<<endl;
    return 0;
}

你可能感兴趣的:((2020牛客暑期多校训练营)[二]Greater and Greater)