NOIP 2002 产生数【建图+DFS+数乘大数】

NOIP 2002 产生数

一种解决思路：可以考虑每一个数字可以最终变换成多少个不同的数字，数字关系可以考虑建成有向图，然后开始深搜广搜都可。

• 注意答案会超过$long$ $long$ $int$

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<string>
#include<map>
#include<queue>
#include<vector>
using namespace std;
#define ll long long int
#define LL ll
#define INF 0x3f3f3f3f
const int maxn = 35;
const int maxe = 1e5 + 10;//最大边数
string n;
int k;
int head[maxe];
struct edge {//链式前向星结构体
	int to, nx;
}e[maxe];
int sz;//总边数
int change_digit[10];//存储每个数字可以变换出的数字的个数（包括本身）
int vis[10];//用于dfs临时记录
void add(int u, int v) {//链式前向星加边
	e[++sz].to = v;
	e[sz].nx = head[u];
	head[u] = sz;
}
void init() {//初始化
	memset(head, -1, sizeof head);//初始化所有头结点为-1
	sz = 0;
}
int dfs(int x) {
	int res = 1;
	for (int i = head[x]; i != -1; i = e[i].nx) {
		int v = e[i].to;
		if (!vis[v]) {
			vis[v] = 1;
			res += dfs(v);
		}
	}
	return res;
}
void multiple_work() {//大数乘法
	int ans[50];
	memset(ans, 0, sizeof ans);
	ans[0] = 1;
	for (int i = 0; i < n.length(); i++) {
		int tmp = n[i] - '0';
		for (int j = 0; j < 50; j++) {
			ans[j] *= change_digit[tmp];//考虑每一位数字可以变换的种数
		}
		for (int j = 0; j < 49; j++) {
			ans[j + 1] += ans[j] / 10;
			ans[j] %= 10;
		}
	}
	int flag = 0;
	for (int i = 50 - 1; i >= 0; i--) {
		if (ans[i]) flag = 1;
		if (flag) cout << ans[i];
	}
	cout << endl;
}
int main() {
	init();
	cin >> n;
	scanf("%d", &k);
	for (int i = 1; i <= k; i++) {//把所有的关系存储进图
		int u, v;
		scanf("%d %d", &u, &v);
		add(u, v);
	}
	for (int i = 0; i <= 9; i++) {
		memset(vis, 0, sizeof vis);
		vis[i] = 1;
		change_digit[i] = dfs(i);
		//cout << change_digit[i] << endl;
	}
	multiple_work();
	return 0;
}
//_CRT_SECURE_NO_WARNINGS