这是一道链表操作的题目,要求把小于x的元素按顺序放到链表前面。我们仍然是使用链表最常用的双指针大法,一个指向当前小于x的最后一个元素,一个进行往前扫描。如果元素大于x,那么继续前进,否则,要把元素移到前面,并更新第一个指针。这里有一个小细节,就是如果不需要移动(也就是已经是接在小于x的最后元素的后面了),那么只需要继续前进即可。算法时间复杂度是O(n),空间只需要几个辅助变量,是O(1)。代码如下:
public ListNode partition(ListNode head, int x) { if(head == null) return null; ListNode helper = new ListNode(0); helper.next = head; ListNode walker = helper; ListNode runner = helper; while(runner.next!=null) { if(runner.next.val<x) { if(walker!=runner) { ListNode next = runner.next.next; runner.next.next = walker.next; walker.next = runner.next; runner.next = next; } else runner = runner.next; walker = walker.next; } else { runner = runner.next; } } return helper.next; }这道题思路比较清晰,不过还是有点细节的,第一次写可能不容易完全写对,可以练习练习。
没有评论:
发表评论