ACM-ICPC 2017 西安赛区现场赛 K. LOVER II && LibreOJ#6062. 「2017 山东一轮集训 Day2」Pair（线段树）

题目链接：西安：https://nanti.jisuanke.com/t/20759   （计蒜客的数据应该有误，题目和 LOJ 的大同小异，题解以 LOJ 为准）

　　　　    LOJ：https://loj.ac/problem/6062

题意：给出一个长度为n的数列a_i和一个长度为m的数列b_i求a_i有多少个长度为m的连续子数列能与b_i匹配。 两个数列可以匹配，当且仅当存在一种方案，使两个数列中的数可以两两配对，两个数可以配对当且仅当它们的和不小于h。

题解：先对 b 数组进行排序，建一颗线段树，将 b 数组中位置为 i 的数初始化为 -i ，那么当 [1,m] 中最小值大于等于 0 时即可匹配。枚举 a 的左端点，同时记录满足条件的最小右端点，当右端点 - 左端点 = m 时子区间满足条件 ans++。（西安现场赛的题则记录左端点可以匹配的最小右端点即可）

 #include <bits/stdc++.h>
 using namespace std;
 #define ll long long
 #define ull unsigned long long
 #define mst(a,b) memset((a),(b),sizeof(a))
 #define mp(a,b) make_pair(a,b)
 #define pi acos(-1)
 #define pii pair<int,int>
 #define pb push_back
 const int INF = 0x3f3f3f3f;
 const double eps = 1e-;
 const int MAXN = 15e4 + ;
 const int MAXM = 2e6 + ;

 int a[MAXN], b[MAXN];;
 int st[MAXN << ], lazy[MAXN << ];

 void pushup(int rt) {
     st[rt] = min(st[rt << ],st[rt <<  | ]);
 }

 void build(int rt,int l,int r) {
     if(l == r) {
         st[rt] = -l;
         return ;
     }
     int mid = (l + r) >> ;
     build(rt << ,l,mid);
     build(rt <<  | ,mid + ,r);
     pushup(rt);
 }

 void pushdown(int rt) {
     if(lazy[rt]) {
         lazy[rt<<] += lazy[rt], lazy[rt<<|] += lazy[rt];
         st[rt<<] += lazy[rt], st[rt<<|] += lazy[rt];
         lazy[rt] = ;
     }
 }

 void update(int rt,int l,int r,int ql,int qr,int val) {
     if(qr < ql) return ;
     if(ql <= l && qr >= r) {
         st[rt] += val;
         lazy[rt] += val;
         return ;
     }
     pushdown(rt);
     int mid = (l + r) >> ;
     if(ql <= mid) update(rt<<,l,mid,ql,qr,val);
     if(qr > mid) update(rt<<|,mid+,r,ql,qr,val);
     pushup(rt);
 }

 int query(int rt,int l,int r,int ql,int qr) {
     if(ql <= l && qr >= r) return st[rt];
     int mid = (l + r) >> ;
     int ans = 1e9;
     if(ql <= mid) ans = min(ans,query(rt<<,l,mid,ql,qr));
     if(qr > mid) ans = min(ans,query(rt<<|,mid+,r,ql,qr));
     return ans;
 }

 int main() {
 #ifdef local
     freopen("data.txt", "r", stdin);
 //    freopen("data.txt", "w", stdout);
 #endif
     int n,m,h;
     scanf("%d%d%d",&n,&m,&h);
     for(int i = ; i <= m; i++) scanf("%d",&b[i]);
     for(int i = ; i <= n; i++) scanf("%d",&a[i]);
     sort(b + , b +  + m);
     build(,,m);
     int now = , ans = ;
     for(int i = ; i <= n; i++) {
         while(query(,,m,,m) <  && now < n) {
             int val = a[++now];
             int pos = lower_bound(b + , b +  + m, h - val) - b;
             update(,,m,pos,m,);
         }
         if(query(,,m,,m) >=  && now == i + m - ) ans++;
         int pos = lower_bound(b + , b +  + m, h - a[i]) - b;
         update(,,m,pos,m,-);
     }
     printf("%d\n",ans);
     return ;
 }