E. Correct Placement

题目链接:Problem - E - Codeforces

题目大意:有n个高为hi,宽为wi的(1<= i <= n)的矩形,判断是否矩形 i 可以包含 矩形 j。即满足:

  • hj < hi 且 wj < wi
  • wj < hi hj < wi

满足以上之一即可,然后将下标 j 写在 下标 i 上的对应数组。

数据范围:

第一行包含一个整数 t ( 1 ≤ t ≤ 1e4 )--测试用例数。

每个测试用例的第一行包含一个整数 n ( 1 ≤ n ≤ 2⋅1e5 ) -数量。

随后是 n 行,两个整数 hi 和 wi 描述。( 1≤hi, wi ≤ 1e9 )--分别是 i 的高度和宽度。

保证所有测试用例的 n 之和不超过 2⋅1e5 。

考察:双指针, 排序, 贪心

解题思路:首先,根据满足的条件可以看出,矩形可以横着放,竖着放。所以每一个矩形可以维护,最大值为高。然后通过高进行排序, 核心:贪心的想法, 最小的矩形最有可能满足,后面的大矩形匹配, 所以采用双指针技巧,维护一个最小的矩形即可。在高度满足的情况下,即通过比较矩形的宽度,维护最小矩形。 注意:代码中 pos1 指向的是p数组的下标, pos是输入时的下标。

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;

using i64 = long long;
using i128 = __int128;

void solve(){
    int n;
    cin >> n;
    vector<int> a(n+1), b(n+1);
    vector<array<int, 3>> p;
    for(int i=1; i<=n; i++) {
        cin >> a[i] >> b[i];
        if(a[i]<b[i]){
            swap(a[i],b[i]);
        }
        p.push_back({a[i],b[i],i});
    }    
    //排序
    sort(p.begin(), p.end(),[&](auto&x, auto&y){
        if(x[0] < y[0] || x[0] > y[0]){
            return x[0] < y[0];
        }else if(x[0]==y[0]) {
            return x[1] < y[1];
        }
    });
    vector<int> ans(n+1,-1);
    
    int i=0,j=0;
    int pos = -1;//p数组的下标
    int pos1 = -1;//原输入的下标
    while(i<n) {
        while(pos==-1 || (j<n && p[i][0] > p[j][0])) {
            if(pos==-1 || p[pos][1] > p[j][1]) {//维护最小的矩形
                pos1 = p[j][2];
                pos = j;
            }
            j++;
        }
        //判断是否可以更新
        if(pos==-1 || (a[p[i][2]] > a[pos1] && b[p[i][2]] > b[pos1])) {
            ans[p[i][2]] = pos1;
        }
        i++;
    }
    for(int i=1; i<=n; i++) {
        cout << ans[i] << " ";
    }
    cout << "\n";
}

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
    int t = 1;
    cin >> t;
    while(t--) {
        solve();
    }
}

感谢收看与点赞, 欢迎大佬指正。

相关推荐
JK0x071 小时前
代码随想录算法训练营 Day40 动态规划Ⅷ 股票问题
算法·动态规划
Feliz..1 小时前
关于离散化算法的看法与感悟
算法
水蓝烟雨2 小时前
1128. 等价多米诺骨牌对的数量
算法·hot 100
codists2 小时前
《算法导论(第4版)》阅读笔记:p11-p13
算法
Kidddddult4 小时前
力扣刷题Day 43:矩阵置零(73)
算法·leetcode·力扣
大龄Python青年6 小时前
C语言 交换算法之加减法,及溢出防范
c语言·开发语言·算法
啊我不会诶6 小时前
CF每日5题
算法
朱剑君7 小时前
排序算法——基数排序
算法·排序算法
COOCC18 小时前
PyTorch 实战:从 0 开始搭建 Transformer
人工智能·pytorch·python·深度学习·算法·机器学习·transformer
拾忆-eleven9 小时前
C++算法(19):整数类型极值,从INT_MIN原理到跨平台开发实战
数据结构·c++·算法