http://codeforces.com/gym/100971/problem/D
给定你一些城市(直线)的坐标和城市人口大小,要求你输出距离每个城市最近的比他人口大的城市标号(不是坐标哦),
如果没有,则置为-1。
单调栈确实好用,维护一个单调递减的栈,分别从左到右,在从右到左,分别计算离他最近的点。
wa了好多,就是没有注意第三组数据,没有给loc排序(坐标),这样得到的最左和最右很可能都不是不对的。。。
用rmq也能写,我感觉rmq就是动态规划版的线段树 O(∩_∩)O哈哈~。
这几天学一下吧,写完再贴代码。
铭神用的线段树,很强。
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <stack>
#include <vector>
#include <algorithm>
using namespace std;
/*维护一个单调递减的栈。
然后每次,
*/
typedef long long ll;
const int maxn=200007;
struct Node
{ ll loc;
ll v;
int num;
};
int cmp2(Node x,Node y)
{ return x.loc<y.loc;
}
int main()
{ int m;
ll a,b;
Node node[maxn];
scanf("%d",&m);
for(int i=1;i<=m;i++)
{ scanf("%lld%lld",&a,&b);
node[i].loc=a;
node[i].v=b;
node[i].num=i;
}
sort(node+1,node+m+1,cmp2);
stack<Node>s;
vector<Node>left;
vector<Node> right;
left.resize(maxn);
right.resize(maxn);
Node w;
w.v=2e10;w.loc=2e10;
w.num=-1;
s.push(w);
for(int i=1;i<=m;i++)
{ while(node[i].v>s.top().v)
s.pop();
//if(!s.empty())
left[i]=s.top();
s.push(node[i]);
}
while(!s.empty())
s.pop();
node[m+1].v=2e10;
node[m+1].loc=2e10;
node[m+1].num=-1;
s.push(node[m+1]);
//s.push(node[m]);
//right[m]=node[m];
for(int i=m;i>=1;i--)
{ while(node[i].v>s.top().v)
{ s.pop();
}
//if(!s.empty())
right[i]=s.top();
s.push(node[i]);
}
/*for(int i=1;i<=m;i++)
{ cout<<left[i].loc<<" "<<right[i].loc<<endl;
}*/
int aim[maxn];
for(int i=1;i<=m;i++)
{ if(abs(left[i].loc-node[i].loc)==abs(right[i].loc-node[i].loc))
{ if(left[i].v>right[i].v)
aim[node[i].num]=left[i].num;
else
aim[node[i].num]=right[i].num;
}
else if(abs(left[i].loc-node[i].loc)>abs(right[i].loc-node[i].loc))
aim[node[i].num]=right[i].num;
else
aim[node[i].num]=left[i].num;
}
//sort(node+1,node+m+1,cmp1);
for(int i=1;i<=m;i++)
{ cout<<left[i].loc<<" "<<right[i].loc<<endl;
}
return 0;
}