近日,微软新大楼的设计方案正在广泛征集中,其中一种方案格外引人注目。在这个方案中,大楼由 n 栋楼组成,这些楼从左至右连成一排,编号依次为 1 到 n,其中第 i栋楼有 h_i层。每栋楼的每一层为一个独立的 办公区域,可以步行 直达同层相邻楼栋的办公区域,以及 直达同楼栋相邻楼层的办公区域。由于方案设计巧妙,上一层楼、下一层楼、向左右移动到相邻楼栋同层的办公区域均刚好需要 1 分钟。在这些办公区域中,有一些被 核心部门 占用了(一个办公区域内最多只有一个核心部门),出于工作效率的考虑,微软希望核心部门之间的移动时间越短越好。对于一个给定的 最大移动时间 k,大楼的 协同值 定义为:有多少个 核心部门对 之间的移动时间不超过 k。由于大楼门禁的限制,不可以走出整个大楼,也不可以登上天台思考人生。你可以认为在办公区域内的移动时间忽略不计,并且在大楼内总是按照最优方案进行移动。对于一个给定的新大楼设计方案,你能算出方案的协同值么?
输入格式
第一行包含两个正整数 n,k(1≤k≤200020),分别表示大楼的栋数以及最大移动时间。第二行包含 n 个正整数 (1≤hi ≤20),分别表示每栋楼的层数。
接下来一行包含一个正整数 m,表示 核心部门 个数。
接下来 m 行,每行两个正整数 x,y核心部门位于第 x栋第y层。
输入数据保证 m 个核心部门的位置不会重复。
对于简单版本:1≤n,m≤50;
对于中等版本:1≤n≤200000,1≤m≤2000;
对于困难版本:1≤n,m≤200000。
输出格式
输出一个整数,即整个大楼的 协同值。
样例解释
样例对应题目描述中的图,核心部门 1 和核心部门 3 之间的距离为 8>7,因此不能计入答案。
样例输入
5 7
4 1 1 3 1
3
1 4
3 1
4 3
样例输出
2
题解:o(m^2)枚举核心区域,logn查询区间小值。
分情况讨论:当最小值>=min(xi,yj)时,距离=两个方向上的差的加和。否则距离=两个核心区域的楼高-2*min+两个楼之间的距离。
代码:
#include<bits/stdc++.h>
using namespace std;
const int N=200090;
int sum[N<<2];
int n,k,m;
int a[N];
struct node
{
int x,y;
bool operator <(const node &t)const
{
return x<t.x||(x==t.x&&y>t.y);
}
}b[N];
void pushup(int rt)
{
sum[rt]=min(sum[rt<<1],sum[rt<<1|1]);
}
void build(int l,int r ,int rt)
{
if(l==r)
{
sum[rt]=a[l];
return ;
}
int mid=(r+l)>>1;
build(l,mid,rt<<1);
build(mid+1,r,rt<<1|1);
pushup(rt);
}
int query(int l,int r,int ll,int rr,int rt )
{
if(ll<=l&&rr>=r)
{
return sum[rt];
}
int mid=(r+l)>>1;
int mi=100;
if(ll<=mid)
mi=min(mi,query(l,mid,ll,rr,rt<<1));
if(rr>mid)
mi=min(mi,query(mid+1,r,ll,rr,rt<<1|1));
return mi;
}
int main()
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
scanf("%d%d",&b[i].x,&b[i].y);
}
sort(b+1,b+1+m);
build(1,n,1);
int ans=0;
for(int i=1;i<=m;i++)
{
for(int j=i+1;j<=m;j++)
{
int len=query(1,n,b[i].x,b[j].x,1);
// cout<<len<<endl;
if(len>=min(b[i].y,b[j].y))
{
if(b[j].x-b[i].x+max(b[i].y,b[j].y)-min(b[i].y,b[j].y)<=k)
ans++;
}
else
{
if(b[i].y+b[j].y-2*len+b[j].x-b[i].x<=k)
ans++;
}
}
}
printf("%d\n",ans);
}