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;}