Wednesday, September 1, 2010

poj_1833_排列.cpp

Problem Links:


poj1833,

Problem:

排列
Time Limit: 1000MS
Memory Limit: 30000K
Total Submissions: 9723
Accepted: 4297
Description
题目描述:
大家知道,给出正整数n,则1到n这n个数可以构成n!种排列,把这些排列按照从小到大的顺序(字典顺序)列出,如n=3时,列出1 2 3,1 3 2,2 1 3,2 3 1,3 1 2,3 2 1六个排列。

任务描述:
给出某个排列,求出这个排列的下k个排列,如果遇到最后一个排列,则下1排列为第1个排列,即排列1 2 3…n。
比如:n = 3,k=2 给出排列2 3 1,则它的下1个排列为3 1 2,下2个排列为3 2 1,因此答案为3 2 1。
Input
第一行是一个正整数m,表示测试数据的个数,下面是m组测试数据,每组测试数据第一行是2个正整数n( 1 <= n < 1024 )和k(1<=k<=64),第二行有n个正整数,是1,2 … n的一个排列。
Output
对于每组输入数据,输出一行,n个数,中间用空格隔开,表示输入排列的下k个排列。
Sample Input
3
3 1
2 3 1
3 1
3 2 1
10 2 
1 2 3 4 5 6 7 8 9 10
Sample Output
3 1 2
1 2 3
1 2 3 4 5 6 7 9 8 10
Source
qinlu@POJ

Solution:


sort simulation;
问题关键在于:给出一个排列a1a2a3…an后,如何求出下一个刚好比它大的排列?举个例子,1 5 3 2 4的下一个排列就是1 5 3 4 2,1 5 4 3 2的下一个排列就是 2 1 3 4 5.
方法:从排列的倒数第二个数向前查找满足ai
如求1 5 4 3 2的下一个排列,查找ai<5),查找比刚好比a0大的数得到a4(即是2),将a0与a4进行交换,并对a1…a4进行排序,得到下一个排列为2 1 3 4 5.
注意
1、考虑最大的排列的下一个排列是最小的排列,如5 4 3 2 1的下一个排列是1 2 3 4 5;
2、考虑效率,输入输出用scanf和printf,用cin和cout会超时.

Source Code:

//Mon Apr 26 05:13:30 CDT 2010
#include <vector>
#include <list>
#include <map>
#include <set>
#include <deque>
#include <queue>
#include <stack>
#include <bitset>
#include <algorithm>
#include <functional>
#include <numeric>
#include <utility>
#include <sstream>
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <string>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <ctime>

using namespace std;

int cmp ( const void *a , const void *b )
{
    return *(int *)a - *(int *)b;
}

void getNext(int v[], int n)
{
    int i, j;
    if (1 == n)
        return;
    for (i = n - 2; i >= 0; i--)
    {
        if (v[i] < v[i + 1])
            break;
    }
    //Check if it's the inverse order;
    if (i < 0)
    {
        for (j = 0; j < n; j++)
            v[j] = j + 1;
        return;
    }
    //find the min from i+1 to the end;
    int idx = i + 1;
    for (j = i + 2; j < n; j++)
    {
        if (v[j] < v[idx] && v[j] > v[i])
            idx = j;
    }
    //Swap;
    int tmp = v[idx];
    v[idx] = v[i];
    v[i] = tmp;
//  sort(v.begin() + i + 1, v.end());
    qsort(v + i + 1, n-1-i, sizeof(v[0]), cmp);
}

int main(int argc, const char* argv[])
{
//  freopen("input.in", "r", stdin);
//  freopen("output.out", "w", stdout);
    int N;
    scanf("%d", &N);

    for (int ncase = 0; ncase < N; ncase++)
    {
        int m, k;
        scanf("%d%d", &m, &k);
        int v[1025];
        for (int i = 0; i < m; i++)
        {
            scanf("%d", &v[i]);
        }
        for (int i = 0; i < k; i++)
        {
            getNext(v, m);
        }
        for (int i = 0; i < m; i++)
        {
            if (i != 0)
                printf(" ");
            printf("%d", v[i]);
        }
        printf("\n");
    }

//  fclose(stdin);
//  fclose(stdout);
    return 0;
}

No comments :