POJ-1780 Code 欧拉路径

题意：破解一套1-6位长度密码的系统，寻找这样一个序列：对于N位的密码10^N+N-1长度的连续的长为N的串能够枚举完所有的密码。

解法：构图之后直接dfs会超内存，因此需要使用栈来实现。调了很久。

代码如下：

#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;

int N, mod, head[100005];
int idx, lim, edge, top;
int stk[1000100];

struct Edge {
int v, next;
}e[1000100]; // 每个节点只能够引出10条边

char vis[1000100];

void insert(int a, int b) {
e[idx].v = b;
e[idx].next = head[a];
head[a] = idx++;
}

void dfs() {
stk[top++] = 0;
while (1) {
int flag = 0;
int v = stk[top-1]; // 取出栈顶元素
if (top == edge-(N-2)) { // 由于遍历的是点，因此从一个点N个9到N个0最终无法到达，中间也就少了N-2个状态
return;
}
for (int i = head[v]; i != -1; i = e[i].next) {
if (!vis[i]) {
flag = 1;
vis[i] = 1;
stk[top++] = e[i].v;
break;
}
}
if (!flag) {
--top;
}
}
}

int main() {
while (scanf("%d", &N), N) {
if (N == 1) {
printf("0123456789\n");
continue;
}
idx = top = 0;
memset(head, 0xff, sizeof (head));
memset(vis, 0, sizeof (vis));
lim = 1;
for (int i = 1; i < N; ++i) {
lim *= 10;
}
edge = lim * 10;
mod = lim / 10; // mod用来提取后后几位
for (int i = 0; i < lim; ++i) {
for (int j = 9; j >= 0; --j) { // 十个状态转移
insert(i, (i%mod)*10+j);
}
}
dfs();
for (int i = 0; i < top-1; ++i) {
printf("%d", stk[i]/mod);
}
printf("%d", stk[top-1]);
for (int i = 1; i < N; ++i) {
printf("0");
}
puts("");
}
return 0;
}