Codeforces Round #693 (Div. 3) E. Correct Placement
题目链接:https://codeforces.com/contest/1472/problem/E
题目大意
n n n 个人,每个人有高 h h h 和宽 w w w 两个属性。第 j j j 个人能站在第 i i i 个人前面当且仅当 h j < h i a n d w j < w i h_j < h_i and w_j < w_i hj<hiandwj<wi 或 w j < h i a n d h j < w i w_j < h_i and h_j < w_i wj<hiandhj<wi。
解题思路
因为每个人既能竖着也能横着放,我们直接把大的值当 h h h ,小的当 w w w ,方便后面处理。
然后我们把 n n n 个人按身高从小到大排序,那么对于我们当前枚举的人 i i i ,他后面的人身高都大于等于他,肯定不满足,我们可以在身高小于 h i h_i hi 的人中找到最小的 w w w ,设为 w m i n w_{min} wmin ,再把 w m i n w_{min} wmin 跟 w i w_i wi 比较即可,若 w m i n < w i w_{min} < w_i wmin<wi ,我们直接把这个人放在 i i i 前面,否则就说明找不到满足条件的人。
注意 i i i 前面可能存在身高等于 h i h_i hi 的,找 w m i n w_{min} wmin 的时候不能算上这些人。
Code
#include <cstdio>
#include <algorithm>
#include <cstring>
#define MAXN 200005
using namespace std;
struct node {
int h, w;
int id;
} a[MAXN];
int ans[MAXN];
bool cmp1(node& t1, node& t2) {
if(t1.h == t2.h) return t1.w < t2.w;
else return t1.h < t2.h;
}
int main(void)
{
int T; scanf("%d", &T);
while(T--) {
int n; scanf("%d", &n);
for(int i=0;i<n;++i) {
int h, w; scanf("%d%d", &h, &w);
if(h < w) swap(h, w);
a[i].h = h; a[i].w = w;
a[i].id = i+1;
}
sort(a, a+n, cmp1);
memset(ans, -1, (n+1) * sizeof(int));
int p = 0;
int minw = 1e9, mink = -1;
for(int i=0;i<n;++i) {
if(i > 0 && a[i].h > a[i-1].h) {
while(p < i) { // p 到 i-1 的 h 是相等的
if(a[p].w < minw) {
minw = a[p].w;
mink = a[p].id;
}
p++;
}
}
if(minw >= a[i].w) ans[a[i].id] = -1;
else ans[a[i].id] = mink;
}
for(int i=1;i<=n;++i) printf("%d ", ans[i]);
printf("\n");
}
return 0;
}