Re: [閒聊] 每日leetcode

[yam276 (虛構史學家)]

※ 引述《yam276 (虛構史學家)》之銘言：
: ※ 引述《sustainer123 (caster )》之銘言：
: : https://leetcode.com/problems/relative-sort-array
: : 1122. Relative Sort Array
: : 給定兩數列arr1與arr2 arr2的元素不重復且皆存在於arr1
: : 請依照arr2的順序排列arr1的元素
: : 假設有元素不在arr2 請遞增排序
: → yam276: 感覺我脫褲子放屁 我哭了                                 06/11 15:35

來看其他大師變魔術

Code:

impl Solution {
    pub fn relative_sort_array(mut arr1: Vec<i32>, arr2: Vec<i32>) -> Vec<i32>
 {
        let mut idx_arr = [-1; 1001];
        for (n, i) in arr2.into_iter().zip(0..) {
            idx_arr[n as usize] = i;
        }
        arr1.sort_unstable_by(|a, b| {
            use std::cmp::Ordering::*;
            match (idx_arr[*a as usize], idx_arr[*b as usize]) {
                (-1, -1) => a.cmp(b),
                (-1 , _) => Greater,
                (_,  -1) => Less,
                (i,   j) => i.cmp(&j),
            }
        });
        arr1
    }
}


太狠了 堪稱高階魔法師

--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 60.251.123.162 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Marginalman/M.1718092396.A.DCB.html
推 sustainer123: 這三小:(((
推 smart0eddie: 三小
→ DJYOSHITAKA: 我瞎了