计蒜课/ 微软大楼设计方案/中等(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 + 10;
int a[MAXN], x[MAXN], y[MAXN], dis[MAXN][30];

int main(void){
    int n, k, m;
    scanf("%d%d", &n, &k);
    for(int i = 1; i <= n; i ++){
        scanf("%d", &a[i]);
        for(int j = 1; j <= 20; j ++){
            if(j == a[i]) dis[i][j] = dis[i - 1][j] + 1;
            else dis[i][j] = dis[i - 1][j];
        }
    }
    scanf("%d", &m);
    for(int i = 0;  i < m; i ++){
        scanf("%d%d", &x[i], &y[i]);
    }
    int ans = 0;
    for(int i = 0; i < m; i++){
        for(int j = i + 1; 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 = 1; d <= 20; d ++){
                    if(dis[cnt2][d] - dis[cnt1 - 1][d] > 0){
                        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] - 2*d + abs(cnt2 - cnt1);
                    if(cnt <= k) ans ++;
                }
            }
        }
    }
    printf("%d\n", ans);
    return 0;
}