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 = ; i < m; ++ i)
             temp2[i] = temp[i] = A[i];
         sort(temp2, temp2 + m, greater<int>());
         string s;
         for(int i = ; i < m; ++ i)
             s += char('' + 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 = ;
 int A[MAXN];

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

 int main()
 {

     int n, m, temp[MAXN] = {}, temp2[MAXN] = {};
     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 = ; i < m; ++ i)
                 temp2[i] = temp[i] = A[i];
             sort(temp2, temp2 + m, greater<int>());
             string s;
             for(int i = ; i < m; ++ i)
                 s += char('' + 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 ;
 }