LOJ #6062. 「2017 山東一輪集訓 Day2」Pair

Description

給出一個長度爲 \(n\) 的數組 \(a\) 和長度爲 \(m\) 的數組 \(b\) ,求 \(a\) 中有多少個長度爲 \(m\) 的連續子序列能夠和 \(b\) 構成完美匹配 , \(a_i\)\(b_j\) 能夠匹配的條件是 \(a_i+b_j>=h\)

Solution

運用 \(Hall\) 定理 , 對於每一個子集都要滿足相連的點集大於等於這個子集的大小
我們把 \(b\) 數組從小到大排序,那麼 \(ai\) 能夠連的邊一定是 \(b\) 的一段後綴,一段 \(a_i\) 的交就是後綴的長度取 \(min\)
那麼我們依次考慮後綴的每一個長度 \(i\) ,設 \(sum_i\) 表示後綴長度小於 \(i\) 的個數,那麼要滿足 \(sum_i<=i\)
所以用線段樹維護一個 \(min(i-sum_i)\) ,判斷一下是否不小於 \(0\) 就好了

#include<bits/stdc++.h>
#define ls (o<<1)
#define rs (o<<1|1)
using namespace std;
const int N=150010;
int n,m,H,a[N],b[N],tr[N*4],la[N*4];
inline void pushdown(int o){
    if(!la[o])return ;
    tr[ls]+=la[o];tr[rs]+=la[o];la[ls]+=la[o];la[rs]+=la[o];la[o]=0;
}
inline int qry(int l,int r,int o,int sa,int se){
    if(sa<=l && r<=se)return tr[o];
    int mid=(l+r)>>1;pushdown(o);
    if(se<=mid)return qry(l,mid,ls,sa,se);
    if(sa>mid)return qry(mid+1,r,rs,sa,se);
   return min(qry(l,mid,ls,sa,mid),qry(mid+1,r,rs,mid+1,se));
}
inline void ins(int l,int r,int o,int sa,int se,int t){
    if(sa<=l && r<=se){tr[o]+=t;la[o]+=t;return ;}
    int mid=(l+r)>>1;pushdown(o);
    if(se<=mid)ins(l,mid,ls,sa,se,t);
    else if(sa>mid)ins(mid+1,r,rs,sa,se,t);
    else ins(l,mid,ls,sa,mid,t),ins(mid+1,r,rs,mid+1,se,t);
    tr[o]=min(tr[ls],tr[rs]);
}
inline void build(int l,int r,int o){
    if(l==r){tr[o]=l;return ;}
    int mid=(l+r)>>1;
    build(l,mid,ls);build(mid+1,r,rs);
    tr[o]=min(tr[ls],tr[rs]);
}
int main(){
    freopen("pp.in","r",stdin);
    freopen("pp.out","w",stdout);
    cin>>n>>m>>H;
    for(int i=1;i<=m;i++)scanf("%d",&b[i]);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    sort(b+1,b+m+1);
    for(int i=1;i<=n;i++)a[i]=m-(lower_bound(b+1,b+m+1,H-a[i])-b)+1;
    build(0,m,1);
    int ans=0;
    for(int i=1;i<=n;i++){
        ins(0,m,1,a[i],m,-1);
        if(i>m)ins(0,m,1,a[i-m],m,1);
        if(i>=m && tr[1]>=0)ans++;
    }
    cout<<ans<<endl;
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章