nyoj 32-组合数(next_permutation, stack, set)

32-组合数

内存限制:64MB 时间限制:3000ms Special Judge: No
accepted:8 submit:11

题目描述:

找出从自然数1、2、... 、n（0<n<10）中任取r(0<r<=n)个数的所有组合。

输入描述:

输入n、r。

输出描述:

按特定顺序输出所有组合。
特定顺序：每一个组合中的值从大到小排列，组合之间按逆字典序排列。

样例输入:

复制

5 3

样例输出:

543
542
541
532
531
521
432
431
421
321

分析：
　　①、要求从n个元素中选m个元素，其实就是从全排列中选择它的前m项就行了；
　　②、因为本题对顺序做了限制，所以我们还要将选出的元素排序、去重；

步骤：
　　①、首先来说，我们要产生一组全排列，这里我们通过STl中的next_permutation
　　②、将每个全排列前m项取出，排序后插入set(PS:set具有去重的功能)
　　③、将从set取出的每一个值依次放入栈（stack）中（利用栈的“先进后出”）
　　④、从栈（stack）中取出、输出

核心代码：
　　

do
{
    if(!equal(A, A + m, temp)) // 解释：如果数组A的前m个元素与temp两个数组不相等
    {
        for(int i = 0; i < m; ++ i)
            temp2[i] = temp[i] = A[i];
        sort(temp2, temp2 + m, greater<int>());
        string s;
        for(int i = 0; i < m; ++ i)
            s += char('0' + temp2[i]);
        my_set.insert(s);
    }
}
while(next_permutation(A, A + n));

C/C++代码实现（AC）：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <stack>
#include <map>
#include <queue>
#include <set>

using namespace std;
const int MAXN = 12;
int A[MAXN];

void cal_array(int n)
{
    for(int i = 0; i < n; ++ i)
        A[i] = i + 1;
}

int main()
{

    int n, m, temp[MAXN] = {0}, temp2[MAXN] = {0};
    scanf("%d%d", &n, &m);
    set <string> my_set;
    set <string> ::iterator iter;
    stack <string> my_stack;
    cal_array(n);
    do
    {
        if(!equal(A, A + m, temp)) // 解释：如果数组A的前m个元素与temp两个数组不相等
        {
            for(int i = 0; i < m; ++ i)
                temp2[i] = temp[i] = A[i];
            sort(temp2, temp2 + m, greater<int>());
            string s;
            for(int i = 0; i < m; ++ i)
                s += char('0' + temp2[i]);
            my_set.insert(s);
        }
    }
    while(next_permutation(A, A + n));

    for(iter = my_set.begin(); iter != my_set.end(); ++ iter)
        my_stack.push(*iter);
    while(!my_stack.empty())
    {
        cout <<my_stack.top() <<endl;;
        my_stack.pop();
    }
    return 0;
}