难度比较篮球杯的离线
题意:
思路:
一开始想的是把询问按h排序,把高度也按val排序,这样一个指针扫完就可以了
事实上确实是这么做的,但是我却在答案统计上出了问题,因为岛屿数量和位置有关
然后想的是,既然有关,那就不能把a数组排序
手摸了一下发现,淹没山峰的贡献是-1,淹没山谷的贡献是1
因此统计了一下每个深度山峰和山谷的数量,然后离线
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int mxn=5e4+10;
const int mxe=5e4+10;
struct ty{
int id,h;
}q[mxn];
struct ty2{
int val,type;
}a[mxn];
map<int,int> mp1,mp2;
int N,Q;
int ans[mxn];
bool cmp(ty x,ty y){
return x.h<y.h;
}
void solve(){
cin>>N>>Q;
a[0].val=-1e9,a[N+1].val=-1e9;
for(int i=1;i<=N;i++) cin>>a[i].val;
for(int i=1;i<=N;i++){
if(a[i].val>a[i-1].val&&a[i].val>a[i+1].val) mp1[a[i].val]++;
else if(a[i].val<a[i-1].val&&a[i].val<a[i+1].val) mp2[a[i].val]++;
}
for(int i=1;i<=Q;i++){
cin>>q[i].h;
q[i].id=i;
}
sort(q+1,q+1+Q,cmp);
int res=1;
for(int i=1;i<=Q;i++){
res+=mp1[q[i].h]*-1;
res+=mp2[q[i].h]*1;
ans[q[i].id]=res;
}
for(int i=1;i<=Q;i++) cout<<ans[i]<<'\n';
}
signed main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int __=1;//cin>>__;
while(__--)solve();return 0;
}
然后就wa了
正解是这两种解法的结合版
在单指针扫描的基础上进行答案统计就行了
怎么统计?记录一下位置就好了,vis记录是否被淹没,如果两边都被淹没过了,那么贡献-1,否则贡献+1
Code:
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int mxn=5e4+10;
const int mxe=5e4+10;
struct ty{
int val,pos;
}a[mxn];
struct ty2{
int id,h;
}q[mxn];
int N,Q;
int vis[mxn],ans[mxn];
bool cmp1(ty x,ty y){
return x.val<y.val;
}
bool cmp2(ty2 x,ty2 y){
return x.h<y.h;
}
void solve(){
cin>>N>>Q;
for(int i=1;i<=N;i++){
cin>>a[i].val;
a[i].pos=i;
}
for(int i=1;i<=Q;i++){
cin>>q[i].h;
q[i].id=i;
}
sort(a+1,a+1+N,cmp1);
sort(q+1,q+1+Q,cmp2);
int idx=1,res=1,mx=0;
vis[0]=1,vis[N+1]=1;
for(int i=1;i<=Q;i++){
if(mx>=q[i].h){
ans[q[i].id]=res;
}else{
mx=q[i].h;
while(idx<=N&&a[idx].val<=q[i].h){
if(vis[a[idx].pos]){
idx++;
continue;
}
if(!vis[a[idx].pos-1]&&!vis[a[idx].pos+1]) res++;
else if(vis[a[idx].pos-1]&&vis[a[idx].pos+1]) res--;
vis[a[idx].pos]=1;
idx++;
}
ans[q[i].id]=res;
}
}
for(int i=1;i<=Q;i++) cout<<ans[i]<<'\n';
}
signed main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int __=1;//cin>>__;
while(__--)solve();return 0;
}