2017 计蒜之道 初赛 第六场 微软大楼设计方案（中等）

网址：https://nanti.jisuanke.com/t/15773
近日，微软新大楼的设计方案正在广泛征集中，其中一种方案格外引人注目。在这个方案中，大楼由 nn 栋楼组成，这些楼从左至右连成一排，编号依次为 11 到 nn，其中第 ii 栋楼有 h_ih
​i
​​ 层。每栋楼的每一层为一个独立的 办公区域，可以步行 直达同层相邻楼栋的办公区域，以及 直达同楼栋相邻楼层的办公区域。

由于方案设计巧妙，上一层楼、下一层楼、向左右移动到相邻楼栋同层的办公区域均刚好需要 11 分钟。在这些办公区域中，有一些被 核心部门 占用了（一个办公区域内最多只有一个核心部门），出于工作效率的考虑，微软希望核心部门之间的移动时间越短越好。对于一个给定的 最大移动时间 kk，大楼的 协同值 定义为：有多少个 核心部门对 之间的移动时间不超过 kk。由于大楼门禁的限制，不可以走出整个大楼，也不可以登上天台思考人生。你可以认为在办公区域内的移动时间忽略不计，并且在大楼内总是按照最优方案进行移动。

对于一个给定的新大楼设计方案，你能算出方案的协同值么？

输入格式

第一行包含两个正整数 n,k(1\leq k\leq 200020)n,k(1≤k≤200020)，分别表示大楼的栋数以及最大移动时间。

第二行包含 nn 个正整数 h_1,h_2,…,h_n(1\leq h_i\leq 20)h
​1
​​ ,h
​2
​​ ,…,h
​n
​​ (1≤h
​i
​​ ≤20)，分别表示每栋楼的层数。

接下来一行包含一个正整数 mm，表示 核心部门 个数。

接下来 mm 行，每行两个正整数 x_i,y_i(1\leq x_i\leq n,1\leq y_i\leq h_{x_i})x
​i
​​ ,y
​i
​​ (1≤x
​i
​​ ≤n,1≤y
​i
​​ ≤h
​x
​i
​​
​​ )，表示该核心部门位于第 x_ix
​i
​​ 栋楼的第 y_iy
​i
​​ 层。

输入数据保证 mm 个核心部门的位置不会重复。

对于简单版本：1\leq n,m\leq 501≤n,m≤50；

对于中等版本：1\leq n\leq 200000,1\leq m\leq 20001≤n≤200000,1≤m≤2000；

对于困难版本：1\leq n,m\leq 2000001≤n,m≤200000。

输出格式

输出一个整数，即整个大楼的 协同值。

样例解释

样例对应题目描述中的图，核心部门 11 和核心部门 33 之间的距离为 8>78>7，因此不能计入答案。

样例输入

5 7
4 1 1 3 1
3
1 4
3 1
4 3
样例输出

2

---

先存一下代码，线段树。等过了困难的再回来写题解。。。
（只能过中等）

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<cmath>
using namespace std;
/******************************************************/
#define LL long long int
#define mem(a,b) memset(a,b,sizeof(a))
#define mid ((l+r)/2)
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define L rt<<1
#define R rt<<1|1
#define N 200000+50
#define pow(a) a*a
#define INF 0x3f3f3f3f
#define max(a,b) (a>b?a:b)
#define min(a,b) (a<b?a:b)
/*********************************************************/
int n, MAX;
int m;
int dat[N];
struct node{
    int l, r, minn;
};
struct tree{
    int x, high;
}house[N];
node s[N<<2];
void build(int l, int r, int rt){
    if (l == r){
        s[rt].minn = dat[l];
        return;
    }
    build(lson);
    build(rson);
    s[rt].minn = (s[L].minn < s[R].minn ? s[L].minn : s[R].minn);
}
int query(int a, int b, int l, int r, int rt){
    if (a <= l&&r <= b){
        return s[rt].minn;
    }
    int m = (l + r) / 2;
    int res = 0, res2 = 0;
    if (a <= m)res = query(a, b, lson);
    if (b > m)res2 = query(a, b, rson);
    return min(res, res2);
}
int dis(tree a, tree b){
    return abs(a.x - b.x) + abs(a.high - b.high);
}
int main(){
    scanf("%d%d", &n, &MAX);
    for (int i = 1; i <= n; i++){
        scanf("%d", &dat[i]);
    }
    build(1, n, 1);
    scanf("%d", &m);
    for (int i = 1; i <= m; i++){
        int a, b;scanf("%d%d", &a, &b);
        house[i].x = a;
        house[i].high = b;
    }
    int cnt = 0;
    for (int i = 1; i <= m; i++){
        for (int j = 1+i; j <= m; j++){
            int did = dis(house[i], house[j]);
            if (did > MAX)continue;
            int minn = query(house[i].x, house[j].x, 1, n, 1);
            if (minn == min(house[i].high, house[j].high)){
                cnt++;
            }
            else{
                int d = house[i].high + house[j].high + abs(house[i].x - house[j].x)-2;
                if (d <= MAX)cnt++;
            }
        }
    }
    cout << cnt  << endl;

}