小白C语言Leetcode————86.分隔链表

题目如下

给定一个链表和一个特定值 x，对链表进行分隔，使得所有小于 x 的节点都在大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

示例:

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/partition-list
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

#include <stdio.h>
#include <stdlib.h>

#define LEN sizeof(struct listnode)
typedef struct listnode{
               int data;
               struct listnode *next;
               }listnode,*linklist;
int main()
{
    linklist creat(),partition(linklist head, int k);
    linklist a,b;
    int x;
    printf("input the linklist:");
    a=creat();
    printf("Input the k:");
    scanf("%d",&x);
    b=partition(a,x);
    printf("Output the answer:");
    while(b)
    {
        printf("%d\t",b->data);
        b=b->next;
    }
    return 0;
}

linklist partition(linklist head,int k)
{
    linklist L1=(linklist)malloc(sizeof(LEN));
    linklist L2=(linklist)malloc(sizeof(LEN));
    L1->next=head;
    linklist cur=head,pre=L1,dummy=L2;
    while(cur->next!=NULL)
    {
        if(cur->data<k)
        {
            pre=pre->next;
            cur=cur->next;
        }
        else
        {
            L2->next=cur;
            cur=cur->next;
            pre->next=cur;
            L2=L2->next;
        }
    }
    cur->next=dummy->next;
    L2->next=NULL;
    return head;
}

linklist creat(linklist L)
{
    linklist s,r;
    int x;
    L=(linklist)malloc(sizeof(LEN));
    r=L;
    scanf("%d",&x);
    while(x!=9999)
    {
        s=(linklist)malloc(sizeof(LEN));
        s->data=x;
        r->next=s;
        r=s;
        scanf("%d",&x);
    }
    r->next=NULL;
    return L->next;
}