codevs 1009 产生数 floyd算法

题目：codevs1009 产生数

思路：一开始没看清题目，以为就是简单地并查集，结果一直错，仔细看了看例子才发现我太天真的了，这题是用Floyd算法求点到点之间的连通关系，可以看成是一张10*10的图：

	0	1	2	3	4	5	6	7	8	9
0	0	0	0	0	0	0	0	0	0	0
1	0	0	0	0	0	0	0	0	0	0
2	0	0	0	0	0	0	0	0	0	0
3	0	0	0	0	0	0	0	0	0	0
4	0	0	0	0	0	0	0	0	0	0
5	0	0	0	0	0	0	0	0	0	0
6	0	0	0	0	0	0	0	0	0	0
7	0	0	0	0	0	0	0	0	0	0
8	0	0	0	0	0	0	0	0	0	0
9	0	0	0	0	0	0	0	0	0	0

一开始每个数都没有变化，所以之间都不连通，然后每输入一个变化情况，对应的两个点连通，之间的权值设为1；然后用floyd求每个点的连通数；也就是每个数能变成的其他数的个数，再遍历给出的整数的每位数，遇到能变换的数让其能变换的情况数相乘即可求出正确答案。

这里由于给出的数会超出整数范围，所以用字符串保存整数。也方便操作：

学习了一波floyd算法：坐在马桶上看算法:只有五行的Floyd最短路算法 - 51CTO.COM

代码如下：

#include<iostream>
#include<set>
#include<map>
#include<string>
#include<cstdio>
#include<cstring>
using namespace std;
string s;
int k;
int a[20][2];
int b[20];
int m[11][11];
long long int count = 1;
int f[20];
void floyd()
{
	for(int i = 0;i <= 9;i++)
	{
		for(int k = 0;k <= 9;k++)
		{
			for(int j = 0;j <= 9;j++)
			{
				if(i!=j && i!=k && j!=k)
				{
					if(m[k][i]&&m[i][j])
					{
						m[k][j] = 1;
					}
				}
			}
		}
	}
}
int main()
{
	getline(cin, s,' ');
	cin >> k;
	//cout << s.size() << endl; 
	memset(m,0,sizeof(m));
	for(int i = 0;i < k;i++)
	{
		cin >> a[i][0] >> a[i][1];
		m[a[i][0]][a[i][1]] = 1;
	}
	floyd();
	for(int i = 0;i <= 9;i++)
	{
		b[i] = 1;
		for(int j = 0;j <= 9;j++)
		{
			if(m[i][j])b[i]++;
		}
	}
	for(int i = 0;i < s.size();i++)
	{
	 	if(b[s[i]-'0'] > 0)count *= (b[s[i]-'0']);
	}
	cout << count;
	return 0;
}