将链表分类成三色旗

import java.security.Policy;

public class SmallerEqualBigger {
  public class Node {
    public int value;
    public Node next;

    Node(int data) { this.value = data; }
  }
  //方法一:将它们存到数组中排序
  public static Node listPartition1(Node head, int pivot, Node tail) {
    if (head == null)
      return head;
    Node cur = head;
    int i = 0;
    while (cur != null) {
      i++;
      cur = cur.next;
    }
    Node[] nodeArr = new Node[i];
    i = 0;
    cur = head;
    for (i = 0; i != nodeArr.length; i++) {
      nodeArr[i] = cur;
      cur = cur.next;
    }
    arrPartion(nodeArr, pivot);
    for (i = 1; i <= nodeArr.length; i++) {
      nodeArr[i - 1].next = nodeArr[i];
    }
    nodeArr[nodeArr.length - 1].next = null;
    return nodeArr[0];
  }
  //三取样切分快排
  public arrPartion(Node[] nodeArr, int pivot) {
    int small = -1;
    int big = nodeArr.length;
    int index = 0;
    while (index != big) {
      if (nodeArr[index].value < pivot) {
        swap(nodeArr, ++small, index++);
      } else if (nodeArr[index].value == pivot) {
        index++;
      } else {
        swap(nodeArr, --big, index);
      }
    }
  }

  public static void swap(Node[] nodeArr, int a, int b) {
    Node tmp = nodeArr[a];
    nodeArr[a] = nodeArr[b];
    nodeArr[b] = tmp;
  }

  //第二种方法:使用少量额外空间
  public static void listPartition2(Node head, int pivot) {
    Node sH = null;   // small head
    Node sT = null;   // small tail
    Node eH = null;   // equal head
    Node eT = null;   // equal tail
    Node bH = null;   // big head
    Node bT = null;   // big tail
    Node next = null; // save
    while (head != null) {
      next = head.next;
      head.next = null;
      if (head.value < pivot) {
        if (sH == null) {
          sH = head;
          sT = head;
        } else {
          sT.next = head;
          sT = head;
        }
      } else if (head.value == pivot) {
        if (eH == null) {
          eH = head;
          eT = head;
        } else {
          eT.next = head;
          eT = head;
        }
      } else {
        if (bH == null) {
          bH = head;
          bT = head;
        } else {
          bT.next = head;
          bT = head;
        }
      }
      head = next;
    }
    // reconnect

    if (sT != null) {
      sT.next = eH;
      eT = eT == null ? sT : eT;
    }
    if (eT != null) {
      eT.next = bH;
    }
    return sH != null ? sH : eH != null ? eH : bH;
  }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值