Check if an array can be formed by merging 2 non-empty permutations
Last Updated :
23 Aug, 2022
Given an array arr[] of length N, the task is to check if it can be formed by merging two permutations of the same or different lengths. Print YES if such merging is possible. Otherwise, print NO.
Permutations of length 3 are {1, 2, 3}, {2, 3, 1}, {1, 3, 2}, {3, 1, 2}, {3, 2, 1}, {2, 1, 3}.
Examples:
Input: arr = [1, 3, 2, 4, 3, 1, 2]
Output: YES
Explanation:
The given array can be formed by merging a permutation of length 4 [1, 3, 2, 4] and permutation of length 3 [3, 1, 2]
Input: arr = [1, 2, 3, 2, 3, 2, 1]
Output: NO
Approach :
We can observe that the minimum excludant (MEX) of a permutation of length N is N+1.
So, if the length of the first permutation is l, then MEX of the prefix arr [0 ...... l-1] is l+1 and the MEX of the suffix a[l ...... n] will be N - l + 1.
So, we can calculate MEX of prefix and suffixes and if the above condition is satisfied, the answer will be "YES". Otherwise, the answer will be "NO".
Below is the implementation of the above approach:
C++
// C++ program for the
// above approach
#include<bits/stdc++.h>
using namespace std;
void if_merged_permutations(int a[],
int n)
{
int pre_mex[n];
// Calculate the mex of the
// array a[0...i]
int freq[n + 1];
memset(freq, 0, sizeof(freq));
for(int i = 0; i < n; i++)
{
pre_mex[i] = 1;
}
// Mex of empty
// array is 1
int mex = 1;
// Calculating the frequency
// of array elements
for(int i = 0; i < n; i++)
{
freq[a[i]]++;
if(freq[a[i]] > 1)
{
// In a permutation
// each element is
// present one time,
// So there is no chance
// of getting permutations
// for the prefix of
// length greater than i
break;
}
// The current element
// is the mex
if(a[i] == mex)
{
// While mex is present
// in the array
while(freq[mex] != 0)
{
mex++;
}
}
pre_mex[i] = mex;
}
int suf_mex[n];
for(int i = 0; i < n; i++)
{
suf_mex[i] = 1;
}
// Calculate the mex of the
// array a[i..n]
memset(freq, 0, sizeof(freq));
// Mex of empty
// array is 1
mex = 1;
// Calculating the frequency
// of array elements
for(int i = n - 1; i > -1; i--)
{
freq[a[i]]++;
if(freq[a[i]] > 1)
{
// In a permutation each
// element is present
// one time, So there is
// no chance of getting
// permutations for the
// suffix of length lesser
// than i
continue;
}
// The current element is
// the mex
if(a[i] == mex)
{
// While mex is present
// in the array
while(freq[mex] != 0)
{
mex++;
}
}
suf_mex[i] = mex;
}
// Now check if there is atleast
// one index i such that mex of
// the prefix a[0..i]= i +
// 2(0 based indexing) and mex
// of the suffix a[i + 1..n]= n-i
for(int i = 0; i < n - 1; i++)
{
if(pre_mex[i] == i + 2 &&
suf_mex[i + 1] == n - i)
{
cout << "YES" << endl;
return;
}
}
cout << "NO" << endl;
}
// Driver code
int main()
{
int a[] = {1, 3, 2,
4, 3, 1, 2};
int n = sizeof(a)/ sizeof(a[0]);
if_merged_permutations(a, n);
}
//This code is contributed by avanitrachhadiya2155
// Java program for above approach
import java.util.*;
import java.lang.*;
import java.io.*;
class GFG{
static void if_merged_permutations(int a[],
int n)
{
int[] pre_mex = new int[n];
// Calculate the mex of the
// array a[0...i]
int[] freq = new int[n + 1];
for(int i = 0; i < n; i++)
{
pre_mex[i] = 1;
}
// Mex of empty
// array is 1
int mex = 1;
// Calculating the frequency
// of array elements
for(int i = 0; i < n; i++)
{
freq[a[i]]++;
if (freq[a[i]] > 1)
{
// In a permutation
// each element is
// present one time,
// So there is no chance
// of getting permutations
// for the prefix of
// length greater than i
break;
}
// The current element
// is the mex
if (a[i] == mex)
{
// While mex is present
// in the array
while (freq[mex] != 0)
{
mex++;
}
}
pre_mex[i] = mex;
}
int[] suf_mex = new int[n];
for(int i = 0; i < n; i++)
{
suf_mex[i] = 1;
}
// Calculate the mex of the
// array a[i..n]
Arrays.fill(freq, 0);
// Mex of empty
// array is 1
mex = 1;
// Calculating the frequency
// of array elements
for(int i = n - 1; i > -1; i--)
{
freq[a[i]]++;
if (freq[a[i]] > 1)
{
// In a permutation each
// element is present
// one time, So there is
// no chance of getting
// permutations for the
// suffix of length lesser
// than i
continue;
}
// The current element is
// the mex
if (a[i] == mex)
{
// While mex is present
// in the array
while (freq[mex] != 0)
{
mex++;
}
}
suf_mex[i] = mex;
}
// Now check if there is atleast
// one index i such that mex of
// the prefix a[0..i]= i +
// 2(0 based indexing) and mex
// of the suffix a[i + 1..n]= n-i
for(int i = 0; i < n - 1; i++)
{
if (pre_mex[i] == i + 2 &&
suf_mex[i + 1] == n - i)
{
System.out.println("YES");
return;
}
}
System.out.println("NO");
}
// Driver code
public static void main(String[] args)
{
int a[] = { 1, 3, 2, 4, 3, 1, 2 };
int n = a.length;
if_merged_permutations(a, n);
}
}
// This code is contributed by offbeat
# Python3 program for above approach
def if_merged_permutations(a, n):
pre_mex =[1 for i in range(n)]
# Calculate the mex of the
# array a[0...i]
freq =[0 for i in range(n + 1)]
# Mex of empty array is 1
mex = 1
# Calculating the frequency
# of array elements
for i in range(n):
freq[a[i]]+= 1
if freq[a[i]]>1:
# In a permutation
# each element is
# present one time,
# So there is no chance
# of getting permutations
# for the prefix of
# length greater than i
break
# The current element
# is the mex
if a[i]== mex:
# While mex is present
# in the array
while freq[mex]!= 0 :
mex+= 1
pre_mex[i]= mex
suf_mex =[1 for i in range(n)]
# Calculate the mex of the
# array a[i..n]
freq =[0 for i in range(n + 1)]
# Mex of empty array is 1
mex = 1
# Calculating the frequency
# of array elements
for i in range(n-1, -1, -1):
freq[a[i]]+= 1
if freq[a[i]]>1:
# In a permutation each
# element is present
# one time, So there is
# no chance of getting
# permutations for the
# suffix of length lesser
# than i
break
# The current element is
# the mex
if a[i]== mex:
# While mex is present
# in the array
while freq[mex]!= 0 :
mex+= 1
suf_mex[i]= mex
# Now check if there is atleast
# one index i such that mex of
# the prefix a[0..i]= i +
# 2(0 based indexing) and mex
# of the suffix a[i + 1..n]= n-i
for i in range(n-1):
if pre_mex[i]== i + 2 and suf_mex[i + 1]== n-i:
print("YES")
return
print("NO")
a =[1, 3, 2, 4, 3, 1, 2]
n = len(a)
if_merged_permutations(a, n)
// C# program for above approach
using System;
class GFG{
static void if_merged_permutations(int[] a,
int n)
{
int[] pre_mex = new int[n];
// Calculate the mex of the
// array a[0...i]
int[] freq = new int[n + 1];
for(int i = 0; i < n; i++)
{
pre_mex[i] = 1;
}
// Mex of empty
// array is 1
int mex = 1;
// Calculating the frequency
// of array elements
for(int i = 0; i < n; i++)
{
freq[a[i]]++;
if (freq[a[i]] > 1)
{
// In a permutation
// each element is
// present one time,
// So there is no chance
// of getting permutations
// for the prefix of
// length greater than i
break;
}
// The current element
// is the mex
if (a[i] == mex)
{
// While mex is present
// in the array
while (freq[mex] != 0)
{
mex++;
}
}
pre_mex[i] = mex;
}
int[] suf_mex = new int[n];
for(int i = 0; i < n; i++)
{
suf_mex[i] = 1;
}
// Calculate the mex of the
// array a[i..n]
Array.Fill(freq, 0);
// Mex of empty
// array is 1
mex = 1;
// Calculating the frequency
// of array elements
for(int i = n - 1; i > -1; i--)
{
freq[a[i]]++;
if (freq[a[i]] > 1)
{
// In a permutation each
// element is present
// one time, So there is
// no chance of getting
// permutations for the
// suffix of length lesser
// than i
continue;
}
// The current element is
// the mex
if (a[i] == mex)
{
// While mex is present
// in the array
while (freq[mex] != 0)
{
mex++;
}
}
suf_mex[i] = mex;
}
// Now check if there is atleast
// one index i such that mex of
// the prefix a[0..i]= i +
// 2(0 based indexing) and mex
// of the suffix a[i + 1..n]= n-i
for(int i = 0; i < n - 1; i++)
{
if (pre_mex[i] == i + 2 &&
suf_mex[i + 1] == n - i)
{
Console.WriteLine("YES");
return;
}
}
Console.WriteLine("NO");
}
// Driver Code
static void Main()
{
int[] a = { 1, 3, 2, 4, 3, 1, 2 };
int n = a.Length;
if_merged_permutations(a, n);
}
}
// This code is contributed by divyeshrabadiya07
<script>
// JavaScript program for above approach
function if_merged_permutations(a, n)
{
var pre_mex = Array(n).fill(0);
// Calculate the mex of the
// array a[0...i]
var freq = Array(n+1).fill(0);
for(var i = 0; i < n; i++)
{
pre_mex[i] = 1;
}
// Mex of empty
// array is 1
var mex = 1;
// Calculating the frequency
// of array elements
for(var i = 0; i < n; i++)
{
freq[a[i]]++;
if (freq[a[i]] > 1)
{
// In a permutation
// each element is
// present one time,
// So there is no chance
// of getting permutations
// for the prefix of
// length greater than i
break;
}
// The current element
// is the mex
if (a[i] == mex)
{
// While mex is present
// in the array
while (freq[mex] != 0)
{
mex++;
}
}
pre_mex[i] = mex;
}
var suf_mex = Array(n).fill(0);;
for(var i = 0; i < n; i++)
{
suf_mex[i] = 1;
}
// Calculate the mex of the
// array a[i..n]
freq = Array(n).fill(0);
// Mex of empty
// array is 1
mex = 1;
// Calculating the frequency
// of array elements
for(var i = n - 1; i > -1; i--)
{
freq[a[i]]++;
if (freq[a[i]] > 1)
{
// In a permutation each
// element is present
// one time, So there is
// no chance of getting
// permutations for the
// suffix of length lesser
// than i
continue;
}
// The current element is
// the mex
if (a[i] == mex)
{
// While mex is present
// in the array
while (freq[mex] != 0)
{
mex++;
}
}
suf_mex[i] = mex;
}
// Now check if there is atleast
// one index i such that mex of
// the prefix a[0..i]= i +
// 2(0 based indexing) and mex
// of the suffix a[i + 1..n]= n-i
for(var i = 0; i < n - 1; i++)
{
if (pre_mex[i] == i + 2 &&
suf_mex[i + 1] == n - i)
{
document.write("YES");
return;
}
}
document.write("NO");
}
// Driver Code
var a = [1, 3, 2, 4, 3, 1, 2];
var n = a.length;
if_merged_permutations(a, n);
</script>
Time Complexity: O(N)
Auxiliary Space: O(N) since using extra space for array freq
Similar Reads
Check if given Array can be made a permutation of 1 to N by reducing elements by half
Given an array nums[] of size N, the task is to check whether the given array can be converted into a permutation of 1 to N after performing given operations any number of times (may be 0). An operation is defined as: Pick any element of the array say 'x', and replace it with 'x/2'. Note: In a permu
6 min read
Check if Array elements of given range form a permutation
Given an array arr[] consisting of N distinct integers and an array Q[][2] consisting of M queries of the form [L, R], the task for each query is to check if array elements over the range [L, R] forms a permutation or not. Note: A permutation is a sequence of length N containing each number from 1 t
13 min read
Check if an Array is a permutation of numbers from 1 to N
Given an array arr containing N positive integers, the task is to check if the given array arr represents a permutation or not. A sequence of N integers is called a permutation if it contains all integers from 1 to N exactly once. Examples: Input: arr[] = {1, 2, 5, 3, 2} Output: No Explanation: The
15+ min read
Check if an Array is a permutation of numbers from 1 to N : Set 2
Given an array arr containing N positive integers, the task is to check if the given array arr represents a permutation or not. A sequence of N integers is called a permutation if it contains all integers from 1 to N exactly once. Examples: Input: arr[] = {1, 2, 5, 3, 2} Output: No Explanation: The
4 min read
Check if empty spaces of Array can be filled maintaining given relations of adjacent
Given two arrays, array A[] of size N and array B[] of size N-1, array A has all the positive elements and -1 for the empty values and array B has 3 characters {'>', '=', '<'} which indicates the following: '>' means A[i] > A[i+1]'=' means A[i] = A[i+1]'<' means A[i] < A[i+1] The t
12 min read
Check if two arrays can be made equal by reversing any subarray once
Given two arrays A[] and B[] of equal size N, the task is to check whether A[] can be made equal to B[] by reversing any sub-array of A only once. Examples: Input: A[] = {1, 3, 2, 4} B[] = {1, 2, 3, 4} Output: Yes Explanation: The sub-array {3, 2} can be reversed to {2, 3}, which makes A equal to B.
8 min read
Check if Array elements in given range form Permutation by performing given updates
Given an array arr[] consisting of N distinct integers and a 2d array Q[][3] consisting of M queries of the 2 types whose operations are as follows: [1, P, X] update position P with X. [2, L, R] check if array elements over the range [L, R] forms a permutation or not. For each query of the second ty
15+ min read
Find if an array of strings can be chained to form a circle | Set 2
Given an array of strings, find if the given strings can be chained to form a circle. A string X can be put before another string Y in a circle if the last character of X is the same as the first character of Y. Examples: Input: arr[] = {"geek", "king"} Output: Yes, the given strings can be chained.
12 min read
Check if rearranging Array elements can form a Palindrome or not
Given a positive integer array arr of size N, the task is to check if number formed, from any arrangement of array elements, form a palindrome or not. Examples: Input: arr = [1, 2, 3, 1, 2]Output: YesExplanation: The elements of a given array can be rearranged as 1, 2, 3, 2, 1. Since 12321 is a pali
7 min read
Check if all strings of an array can be made same by interchanging characters
Given an array arr[] of size N consisting of equal length strings, the task is to check if it is possible to make all strings of the array can be equal or not by swapping any character of one string with any character of the same string or another string. Note: Perform the operation 0 or more times.
12 min read