1. 程式人生 > 其它 >【基礎演算法】next_permutation的手動實現

【基礎演算法】next_permutation的手動實現

420. 火星人

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 1e4 + 10;

int n, m;
int w[N];

int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i ++ ) scanf("%d", &w[i]);
    while(m -- )
    {
        int k = n;
        while(w[k - 1] > w[k]) k -- ; 
        int t = k;
        while(w[t + 1] > w[k - 1]) t ++ ;
        swap(w[k - 1], w[t]);
        reverse(w + k, w + n + 1);
    }
    for(int i = 1; i <= n; i ++ ) printf("%d ", w[i]);
    return 0;
}

作者:Once.
連結:https://www.acwing.com/activity/content/code/content/3396797/
來源:AcWing
著作權歸作者所有。商業轉載請聯絡作者獲得授權,非商業轉載請註明出處。

next_permutation寫法

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 1e4 + 10;

int n, m;
int w[N];

int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i ++ ) scanf("%d", &w[i]);
    while(m -- )
    {
        next_permutation(w + 1, w + n + 1);
    }
    for(int i = 1; i <= n; i ++ ) printf("%d ", w[i]);
    return 0;
}

作者:Once.
連結:https://www.acwing.com/activity/content/code/content/3396797/
來源:AcWing
著作權歸作者所有。商業轉載請聯絡作者獲得授權,非商業轉載請註明出處。