765. 情侣牵手

N 对情侣坐在连续排列的 2N 个座位上,想要牵到对方的手。 计算最少交换座位的次数,以便每对情侣可以并肩坐在一起。 次交换可选择任意两人,让他们站起来交换座位。

人和座位用 0 到 2N-1 的整数表示,情侣们按顺序编号,第一对是 (0, 1),第二对是 (2, 3),以此类推,最后一对是 (2N-2, 2N-1)

这些情侣的初始座位  row[i] 是由最初始坐在第 i 个座位上的人决定的。

示例 1:

输入: row = [0, 2, 1, 3]
输出: 1
解释: 我们只需要交换row[1]和row[2]的位置即可。

示例 2:

输入: row = [3, 2, 0, 1]
输出: 0
解释: 无需交换座位,所有的情侣都已经可以手牵手了。

说明:

  1. len(row) 是偶数且数值在 [4, 60]范围内。

  1. 可以保证row 是序列 0...len(row)-1 的一个全排列。

我的思路:   情侣成对的情况是2n,2n+1. 那么对于一对位置来说,有四种情况,前奇后偶,前偶后奇,前奇后奇,前偶后偶。

无论是哪一种情况,如果对于一个数来说,它的交换能够同时满足交换双方的需求,那么优先交换这个数,否则随便交换哪一个。

class Solution {
public:
    int minSwapsCouples(vector<int>& row) {
        unordered_map<int, int> hash;   // map from person id to position
        for (int i = 0; i < row.size(); ++i) {
            hash[row[i]] = i;
        }
        int ret = 0;
        for (int i = 0; i < row.size(); i += 2) {
            int p1 = row[i];
            int p2 = p1 % 2 == 0 ? p1 + 1 : p1 - 1; // its couple's id
            if (row[i + 1] != p2) {
                int p2_pos = hash[p2];
                hash[row[i + 1]] = p2_pos;
                hash[p2] = i + 1;
                swap(row[i + 1], row[p2_pos]);
                ++ret;
            }
        }
        return ret;
    }
};

猜你喜欢

转载自blog.csdn.net/xuxuxuqian1/article/details/80726373
今日推荐