ACM学习历程—HDU1716 排列2（dfs && set容器）

Description

Ray又对数字的列产生了兴趣：
现有四张卡片，用这四张卡片能排列出很多不同的4位数，要求按从小到大的顺序输出这些4位数。

Input

每组数据占一行，代表四张卡片上的数字（0<=数字<=9），如果四张卡片都是0，则输入结束。

Output

对每组卡片按从小到大的顺序输出所有能由这四张卡片组成的4位数，千位数字相同的在同一行，同一行中每个四位数间用空格分隔。

每组输出数据间空一行，最后一组数据后面没有空行。

Sample Input

1 2 3 4

1 1 2 3

0 1 2 3

0 0 0 0

Sample Output

1234 1243 1324 1342 1423 1432

2134 2143 2314 2341 2413 2431

3124 3142 3214 3241 3412 3421

4123 4132 4213 4231 4312 4321

1123 1132 1213 1231 1312 1321

2113 2131 2311

3112 3121 3211

1023 1032 1203 1230 1302 1320

2013 2031 2103 2130 2301 2310

3012 3021 3102 3120 3201 3210

题目要求的就是四个数全排列从小到大输出。

由于考虑到需要排序，而且要去重，所以直接使用了set容器。

全排列用dfs生成。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <set>
#define LL long long

using namespace std;

int a[5];
bool vis[5];
set <int> ans;

void dfs(int now, int val)
{
    if (now == 4)
    {
        if (val >= 1000)
            ans.insert(val);
        return;
    }
    for (int i = 0; i < 4; ++i)
    {
        if (vis[i])
            continue;
        vis[i] = true;
        dfs(now+1, val*10+a[i]);
        vis[i] = false;
    }
}

void Work()
{
    ans.clear();
    memset(vis, 0, sizeof(vis));
    dfs(0, 0);
    set <int>::iterator it;
    int high = -1;
    bool flag = false;
    for (it = ans.begin(); it != ans.end(); ++it)
    {
        if (high == -1)
            high = *it/1000;
        if (high != -1 && *it/1000 != high)
        {
            printf("\n");
            high = *it/1000;
            flag = false;
        }
        if (flag)
            printf(" ");
        printf("%d", *it);
        flag = true;
    }
    printf("\n");
}

int main()
{
    //freopen("test.in", "r", stdin);
    bool flag = false;
    while (scanf("%d%d%d%d", &a[0], &a[1], &a[2], &a[3]) != EOF &&
           (a[0]+a[1]+a[2]+a[3]))
    {
        if (flag)
            printf("\n");
        Work();
        flag = true;
    }
    return 0;
}