Code POJ - 1780（栈模拟dfs）

#include <iostream>
#include <cstdio>
#include <sstream>
#include <cstring>
#include <map>
#include <cctype>
#include <set>
#include <vector>
#include <stack>
#include <queue>
#include <algorithm>
#include <cmath>
#include <bitset>
#define rap(i, a, n) for(int i=a; i<=n; i++)
#define rep(i, a, n) for(int i=a; i<n; i++)
#define lap(i, a, n) for(int i=n; i>=a; i--)
#define lep(i, a, n) for(int i=n; i>a; i--)
#define rd(a) scanf("%d", &a)
#define rlld(a) scanf("%lld", &a)
#define rc(a) scanf("%c", &a)
#define rs(a) scanf("%s", a)
#define pd(a) printf("%d\n", a);
#define plld(a) printf("%lld\n", a);
#define pc(a) printf("%c\n", a);
#define ps(a) printf("%s\n", a);
#define MOD 2018
#define LL long long
#define ULL unsigned long long
#define Pair pair<int, int>
#define mem(a, b) memset(a, b, sizeof(a))
#define _  ios_base::sync_with_stdio(0),cin.tie(0)
//freopen("1.txt", "r", stdin);
using namespace std;
const int maxn = 1e6 + 10, INF = 0x7fffffff, LL_INF = 0x7fffffffffffffff;
int n, tot, ss;
int stk[maxn], cnt[maxn], ans[maxn];    //stk存最后的结果   cnt里存的是每层的dfs的i的值  ans里是每层没有取模时的v  
int f[maxn];
void no_dfs(int u)
{
    while(cnt[u] < 10)
    {
        int tmp = u * 10 + cnt[u];
        cnt[u]++;
        ans[tot++] = tmp;      //为什么先记录 在取模呢？ 因为我们需要的是每次的那个“边” ，而这个no_dfs函数的作用是求出来所有层的dfs在同一个i时的值 
        u = tmp % f[n-1];
    }
}

int main()
{
    f[0] = 1;
    for(int i = 1; i <= 6; i++) f[i] = f[i-1] * 10;
    while(cin >> n && n)
    {
        mem(cnt, 0);
        ss = tot = 0;
        no_dfs(0);
        while(tot)
        {
            int x = ans[--tot];
            stk[ss++] = x % 10; //取模既能得到最后一位 也就是“边”
            no_dfs(x / 10);     //相当于dfs中回溯到上一层
        }
        for(int i = 0; i < n - 1; i++)
            cout << "0";
        for(int i = ss - 1; i >= 0; i--)
            cout << stk[i];
        cout << endl;

    }

    return 0;
}