Last Updated: 14 Mar, 2021

Partition List

Easy

Asked in companies

Solve now

Problem statement

You are given a Singly Linked List of integers with a head pointer. Every node of the Linked List has a value written on it.

A sample Linked List

Now you have been given an integer value ‘K’. Your task is to rearrange this linked list such that the following conditions hold true :

1. All nodes that have a value less than ‘K’ must come before the nodes that have a value greater than equal to ‘K’.

2. All nodes must maintain the original relative order as they have present in the original linked list after rearrangement.

Input Format:

The first line of the input contains an integer ‘T’ representing the number of test cases.

The first line of each test case contains space-separated integers denoting the values of nodes of the Linked List. The Linked List is terminated with -1. Hence, -1 is never a node value of the Linked List.

The second line of each test case contains a single integer ‘K’.

Output Format:

For each test case return space-separated integers denoting the values of nodes of the Linked List.

The output of each test case will be printed in a separate line.

Note:

You do not need to print anything, it has already been taken care of. Just implement the given function.

Constraints:

1 <= T <= 5
0 <= N <= 5*10^4
1 <= DATA <= 10^9 and DATA != -1
1 <= K <= 10^9   

Time Limit: 1 sec

Solve now

Approaches

01 Approach

The idea here is to Make two-pointers and store all nodes that have values less than ‘K’ in one pointer and all nodes that have values greater or equal to ‘K’ in another pointer. Then we will merge these two to get our answer.

Algorithm:

Create two nodes that say ‘LESS’ and ‘GREATER’ to keep track of nodes that are less than ‘K’ and nodes that are greater than or equal to ‘K’
Create two nodes, say ‘N1’ and ‘N2’ and initialize ‘N1’ with ‘LESS’ and ‘N2’ with ‘GREATER’.
Run a loop until HEAD != NULL. Here ‘HEAD’ is the head of the given linked list.
If the value of HEAD < K then assign ‘HEAD’ in next of ‘LESS’ i.e. do LESS→NEXT = HEAD and move pointer of ‘LESS’ i.e. do LESS = LESS -> NEXT.
Else assign ‘HEAD’ in next of ‘GREATER’ i.e. do GREATER -> NEXT = HEAD and move the pointer of ‘GREATER’ i.e. do GREATER = GREATER -> NEXT.
Move pointer of ‘HEAD’ i.e. do HEAD = HEAD -> NEXT.
Assign NULL to next of ‘GREATER’ i.e. do GREATER -> NEXT = NULL.
Assign N2 -> NEXT to LESS→NEXT.
Return N1 -> NEXT as this is our required HEAD node.

Solve now