Problem
Given an array A
, we can perform a pancake flip: We choose some positive integer k <= A.length
, then reverse the order of the first k elements of A
. We want to perform zero or more pancake flips (doing them one after another in succession) to sort the array A
.
Return the k-values corresponding to a sequence of pancake flips that sort A
. Any valid answer that sorts the array within 10 * A.length
flips will be judged as correct.
Note:
1 <= A.length <= 100
A[i]
is a permutation of[1, 2, ..., A.length]
Example
1
Input: [3,2,4,1]
Output: [4,2,4,3]
Explanation:
We perform 4 pancake flips, with k values 4, 2, 4, and 3.
Starting state: A = [3, 2, 4, 1]
After 1st flip (k=4): A = [1, 4, 2, 3]
After 2nd flip (k=2): A = [4, 1, 2, 3]
After 3rd flip (k=4): A = [3, 2, 1, 4]
After 4th flip (k=3): A = [1, 2, 3, 4], which is sorted.
2
Input: [1,2,3]
Output: []
Explanation: The input is already sorted, so there is no need to flip anything.
Note that other answers, such as [3, 3], would also be accepted.
Solution
The given sequence is not sorted, if we want to shrink the size of the state, we can arrange the
Code
class Solution {
public:
bool vis[101];
int find_idx(int val,vector<int>& tem,int len)
{
for(int i=0;i<len;i++)
{
if(tem[i]==val)
{
//vis[i]=1;
return i;
}
}
return -1;
}
void rev(vector<int>& tem,int k)
{
for(int i=0;i<(k+1)/2;++i)
{
swap(tem[i],tem[k-i]);
//vis[i]=vis[k-i-1];
}
}
vector<int> pancakeSort(vector<int>& A) {
memset(vis,0,sizeof(vis));
vector<int> res;
auto cpy = A;
sort(cpy.begin(),cpy.end());
int N = A.size();
for(int i=N-1;i>=0;--i)
{
if(A[i]==cpy[i]) continue;
int idx=find_idx(cpy[i],A,i+1);
if(idx+1!=1)
res.push_back(idx+1);
rev(A,idx);
if(i+1!=1)
res.push_back(i+1);
rev(A,i);
}
return res;
}
};