算法练习-巧用双指针

发表于 分类于

看题：

首尾双指针

调整数组顺序使奇数位于偶数前面

输入一个整数数组，实现一个函数来调整该数组中数字的顺序，使得所有奇数位于数组的前半部分，所有偶数位于数组的后半部分。

示例：

1
2
3
输入：nums = [1,2,3,4]
输出：[1,3,2,4] 
注：[3,1,2,4] 也是正确的答案之一。

对于这道题，最直接的想法就是遍历，然后相邻元素判断是否为一奇数一偶数，如果满足则交换，直至到尾部。有没有更好的解法呢？

这道题可以使用一个首尾双指针解决：

  1. 定义头指针 left ，尾指针 right .
  2. left 一直往右移，直到它指向的值为偶数
  3. right 一直往左移， 直到它指向的值为奇数
  4. 交换 nums[left] 和 nums[right] .
  5. 重复上述操作，直到 left == right .

img

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
#include<vector>

class Solution {
public:
    vector<int> exchange(vector<int>& nums) {
        int size = nums.size();
        int left = 0, right = size - 1;
        while(left < right){
            if(nums[left]%2){
                left++;
                continue;
            }
            if(!(nums[right]%2)){
                right--;
                continue;
            }
            swap(nums[left++], nums[right--]);
        }
        return nums;
    }
    
};

快慢双指针

链表中倒数第k个节点

输入一个链表，输出该链表中倒数第k个节点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾节点是倒数第1个节点。例如，一个链表有 6 个节点，从头节点开始，它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。

示例：

1
2
3
给定一个链表: 1->2->3->4->5, 和 k = 2.

返回链表 4->5.

image-20210705164118371

快慢指针，先让快指针走k步，然后两个指针同步走，当快指针走到头时，慢指针就是链表倒数第k个节点。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
class Solution {
public:
    ListNode* getKthFromEnd(ListNode* head, int k) {
        int n = 1;
        ListNode* temp = head;
        while(n < k){
            temp = temp->next;
            n++;
        }
        while(temp->next){
            head = head->next;
            temp = temp->next;
        }
        return head;
    }
};