[leetcode] 765. Couples Holding Hands

题目:

N couples sit in 2N seats arranged in a row and want to hold hands. We want to know the minimum number of swaps so that every couple is sitting side by side. A swap consists of choosing any two people, then they stand up and switch seats.

The people and seats are represented by an integer from 0 to 2N-1, the couples are numbered in order, the first couple being (0, 1), the second couple being (2, 3), and so on with the last couple being (2N-2, 2N-1).

The couples' initial seating is given by row[i] being the value of the person who is initially sitting in the i-th seat.

Example 1:

Input: row = [0, 2, 1, 3]
Output: 1
Explanation: We only need to swap the second (row[1]) and third (row[2]) person.

Example 2:

Input: row = [3, 2, 0, 1]
Output: 0
Explanation: All couples are already seated side by side.

Note:

  1. len(row) is even and in the range of [4, 60].
  2. row is guaranteed to be a permutation of 0...len(row)-1

代码:

#include<algorithm>
#include<vector>
class Solution {
public:
    int minSwapsCouples(vector<int>& row) {
        int cnt = 0;
        for(vector<int>::iterator it = row.begin(); it != row.end(); it+=2){
            int num = (*it) % 2 == 0? (*it) + 1 : (*it) - 1;
            vector<int>::iterator temp = find(it+1, row.end(), num);
            if(temp != it+1){
                swap(*(it+1), *(temp));
                cnt++;
            }
            
        }
        return cnt;
    }
};
class Solution {
public:
    int minSwapsCouples(vector<int>& row) {
        int n = row.size();
        int count = 0;
        vector<int> pos(n);
        for(int i = 0; i < n; ++i)
            pos[row[i]] = i;
        for(int i = 0; i < n; i += 2)
        {
            int couple = row[i] % 2 == 0 ? row[i] + 1 : row[i] - 1;
            if(pos[couple] != i + 1)
            {
                Swap(row, pos, i + 1, pos[couple]);
                ++count;
            }
        }
        return count;
    }
    
    void Swap(vector<int> & row, vector<int> & pos, int pos1, int pos2)
    {
        swap(row[pos1], row[pos2]);
        swap(pos[row[pos1]], pos[row[pos2]]);
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值