洛谷P1037产生数

题目链接：

https://www.luogu.org/problemnew/show/P1037

题目描述

给出一个整数n（n<=2000）和k个变换规则（k≤15）。规则：
① 1个数字可以变换成另1个数字；
② 规则中，右边的数字不能为零。
例如：n=234，k=2规则为
2 → 5
3 → 6
上面的整数234经过变换后可能产生出的整数为（包括原数）234，534，264，564共4种不同的产生数。
求经过任意次的变换（0次或多次），能产生出多少个不同的整数。仅要求输出不同整数个数。
输入格式

n
k
x1 y1
x2 y2
… …
xn yn

输出
格式为一个整数（满足条件的整数个数）。

样例输入:

234
2
2 5
3 6

样例输出:

4

解题思路：

floyd+高精度乘法。
由于每个数字可以转化的状态包括直接转化和间接转化。如对于例子：

236
2
2 5
5 7

那么对于数字2而言，可以转化的包括2本身，2->5，2->7，间接转化可以依靠floyd算法；然后将n中每个数字的转化次数相乘就可以了。由于相乘数字很大，采用高精度乘法。

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
using namespace std;
int cnt[10];//统计0到9每个数字总的转化次数
int vis[10][10];//记录i->j是否可以转化
struct bignum{//高精度
    int num[1000];
    int len;
    void init(){
        memset(num,0,sizeof(num));
        len=0;
    }
};
bignum mul(bignum a,bignum b){//高精度乘法
    bignum ans;ans.init();
     
    ans.len=a.len+b.len;
    for(int i=0;i<a.len;i++){
        for(int j=0;j<b.len;j++){
            ans.num[i+j]+=a.num[i]*b.num[j];
        }
    } 
    for(int i=0;i<ans.len;i++){
        if(ans.num[i]>10){
            ans.num[i+1]+=ans.num[i]/10;
            ans.num[i]%=10;
        }
    }
    while(ans.num[ans.len-1]==0) ans.len--;//消除前导0
    return ans;
}
int main(int argc, char** argv) {	 
    string n;
    int k;
    cin>>n>>k;
    for(int i=0;i<k;i++){
        int a,b;
        scanf("%d %d",&a,&b);
        vis[a][b]=1;
    }
    for(int i=0;i<10;i++) vis[i][i]=1;//自身到自身可以转化
    
    for(int k=1;k<10;k++){
        for(int i=0;i<10;i++){
            for(int j=1;j<10;j++){
                if(!vis[i][j]&&vis[i][k]&&vis[k][j]) vis[i][j]=1; //floyd计算间接转化
            }
        }
    }
    
    for(int i=0;i<10;i++){
        for(int j=0;j<10;j++){
            if(vis[i][j]) cnt[i]++;
        }
//		printf("%d\n",cnt[i]);
    }
    
    bignum ans;ans.num[0]=1;ans.len=1;
    for(int i=0;n[i];i++){
        int now=n[i]-'0';
        int x=cnt[now];
        bignum tmp;tmp.init();
        while(x>0){
            tmp.num[tmp.len++]=x%10;
            x/=10;
        }
        ans=mul(ans,tmp);
    }
    
    for(int i=ans.len-1;i>=0;i--) printf("%d",ans.num[i]);
    printf("\n");
    
    return 0;
}

原bfs搜索方法，TLE
解决方法就是利用广搜，注意每次替换为新的数时的处理。

# include<stdio.h>
# include<string.h>
# include<queue>
using namespace std;
queue<int> Q;

int vis[10000];
struct E {
	int a,b;
} po[16];
int main() {
	int n,k;
	int i;
	while(scanf("%d%d",&n,&k)!=EOF) {
		while(Q.empty()==false)
			Q.pop();
		Q.push(n);
		for(i=0; i<k; i++)
			scanf("%d%d",&po[i].a,&po[i].b);

		int cnt=0;
		while(Q.empty()==false) {
			int x=Q.front();
			if(vis[x]==0) {
				cnt++;
				printf("%d\n",x);
				vis[x]=1;
			}
			int wei=1;
			while(x>0) {
				int tmp=x%10;
				for(i=0; i<k; i++) {
					if(tmp==po[i].a) {
						int nx=Q.front()+(po[i].b-tmp)*wei;
						if(vis[nx]==0) {
							Q.push(nx);
						}
					}
				}
				x/=10;
				wei*=10;
			}
			Q.pop();
		}

		printf("%d\n",cnt);
	}
	return 0;
}