计蒜课/ 微软大楼设计方案/中等(xjb)

题目链接：https://nanti.jisuanke.com/t/15772

题意：中文题诶～

思路：对于坐标为p1(x1, y1), p2(x2, y2) 的两个核心, 其中 x1 <= x2 用 d(p1, p2) 表示两者间最矮的大楼，则需要时间为：

对于d(p1, p2) >= min(y1, y2)情况，cnt = abs(x2 - x1) + abs(y2 - y1)

对于d(p1, p2) < min(y1, y2)情况，cnt = y1 + y2 - 2*d(p1, p2) + abs(x2 -x1)

所以对于中等难度，可以枚举所有核心组合的情况，对于当前情况，若 cnt <= k，则计数加一；

那么现在问题为求 d，可以用 dis[i][j] 存储前 i 个元素 j 出现的次数，那么只需要枚举 h 即可得到 d(p1, p2)，而 h <= 20，显然是可行的．．．

代码：

 #include <iostream>
 #include <stdio.h>
 #include <math.h>
 #include <algorithm>
 #include <string.h>
 using namespace std;

 const int MAXN = 2e5 + ;
 int a[MAXN], x[MAXN], y[MAXN], dis[MAXN][];

 int main(void){
     int n, k, m;
     scanf("%d%d", &n, &k);
     for(int i = ; i <= n; i ++){
         scanf("%d", &a[i]);
         for(int j = ; j <= ; j ++){
             if(j == a[i]) dis[i][j] = dis[i - ][j] + ;
             else dis[i][j] = dis[i - ][j];
         }
     }
     scanf("%d", &m);
     for(int i = ;  i < m; i ++){
         scanf("%d%d", &x[i], &y[i]);
     }
     int ans = ;
     for(int i = ; i < m; i++){
         for(int j = i + ; j < m; j++){
             int cnt1 = x[i], cnt2 = x[j];
             if(cnt1 > cnt2){
                 int cc = cnt1;
                 cnt1 = cnt2;
                 cnt2 = cc;
             }
             if(cnt1 == cnt2){
                 if(abs(y[i] - y[j]) <= k) ans++;
             }else{
                 int d;
                 int cc = min(y[i], y[j]);
                 for(d = ; d <= ; d ++){
                     if(dis[cnt2][d] - dis[cnt1 - ][d] > ){
                         break;
                     }
                 }
                 if(cc <= d){
                     int cnt = abs(cnt2 - cnt1) + abs(y[i] - y[j]);
                     if(cnt <= k) ans ++;
                 }else{
                     int cnt = y[i] + y[j] - *d + abs(cnt2 - cnt1);
                     if(cnt <= k) ans ++;
                 }
             }
         }
     }
     printf("%d\n", ans);
     return ;
 }