leetcode 840 优势洗牌

思路:很简单,田忌赛马。强对强,弱对弱,如果发现你的强打不过对方的强,就用最弱的对对方的强;

程序实现问题:

1、为了分出强弱,肯定要sort()排序一下,

sort(C.rbegin(), C.rend());     降序

sort(C.begin(), C.end());        升序

2、排序后索引都变了,怎么存储索引,

定义  

vector> C; 

 C.push_back({B[i], i});

pair对存值和索引,注意初始化,这不是双vector

3、交换vector中的次序,涉及到

A.insert(A.begin() + i, A[size - 1]);     插入在第i个元素之前
                A.pop_back();

代码:

class Solution {
public:
    vector advantageCount(vector& A, vector& B) {
        vector> C;
        int size = A.size();
        
        vector result(size, -1);
        
        
        for(int i = 0; i < size; i++)
        {
            C.push_back({B[i], i});
        }
        
        sort(C.rbegin(), C.rend());
        sort(A.rbegin(), A.rend());
        
        for(int i = 0; i < size; i++)
        {
            if(A[i] <= C[i].first)
            {
                A.insert(A.begin() + i, A[size - 1]);
                A.pop_back();
                
            }
            
        }
        
        for(int j =0; j< size; j++)
        {
            result[C[j].second] = A[j];
            
        }
        return result;
    }
};

你可能感兴趣的:(leetcode 840 优势洗牌)