D. BFS Trees(bfs+枚举计数)

题目

题意：

    给定一张n个点m条边的联通无权图，要求输出对于任意的i,j，在原图中有多少棵生成树，使得i,j到生成树上的任意一点的距离等于i,j在原图上到达他们的距离。
    $1\le n\le 400,1\le m\le 600$

分析：

    首先我们预处理出所有点的最短路。然后我们考虑将一个点x加入到最小生成树，那连那条边呢？能连的点的距离一定是x与i的距离减一，x与j的距离减一，这样连上去才能保证i,j到x的距离为最短距离。
    还有一种点，他们在i,j的路径上，这时候的距离要求又不相同，所以我们直接找到i到j的一条路径，直接从i开始搜索，找到一个点，这个点离j更近了一步，离i更远了一步，这样就把这条路径处理出来了，其他的点就按照上述的方法加入即可。

#include <iostream>
#include <queue>
#include <vector>
#include <cstring>
using namespace std;

const int maxn = 405;
typedef long long ll;
const int mod = 998244353;

vector<int> g[maxn];
ll dis[maxn][maxn],ans[maxn][maxn],vis[maxn];
int n;

void bfs(int begin)
{
	for (int i = 1; i <= n; i++) dis[begin][i] = 1e18;
	queue<int> q;
	q.push(begin);
	dis[begin][begin] = 0;
	while( !q.empty() )
	{
		int x = q.front();
		q.pop();
		for (int i = 0; i < g[x].size(); i++)
		{
			int t = g[x][i];
			if( dis[begin][t] > dis[begin][x] + 1 )
			{
				dis[begin][t] = dis[begin][x] + 1;
				q.push(t);
			} 
		}
	}
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	int m;
	cin >> n >> m;
	for (int i = 1; i <= m; i++)
	{
		int x,y;
		cin >> x >> y;
		g[x].push_back(y);
		g[y].push_back(x);
	}
	for (int i = 1; i <= n; i++) bfs(i);
	for (int i = 1; i <= n; i++)
	{
		for (int j = 1; j <= i; j++)
		{
			memset(vis,0,sizeof(vis));
			int sx = i,tx = j;
			while( sx != tx )
			{
				vis[sx] = 1;
				for (int k = 0; k < g[sx].size(); k++)
				{
					int t = g[sx][k];
					if( dis[i][sx] + 1 == dis[i][t] && dis[j][sx] == dis[j][t] + 1 )
					{
						sx = t;
						break;
					}
				}
			}
			vis[tx] = 1;
			ans[i][j] = 1;
			for (int k = 1; k <= n; k++)
			{
				if( !vis[k] )
				{
					int cnt = 0;
					for (int l = 0; l < g[k].size(); l++)
					{
						int t = g[k][l];
						if( dis[i][k] == dis[i][t]+1 && dis[j][k] == dis[j][t]+1 )
							cnt ++;
					}
					vis[k] = 1;
					ans[i][j] = ans[i][j] * cnt % mod;
				}
			}
			ans[j][i] = ans[i][j];
		}
	} 
	for (int i = 1; i <= n; i++)
	{
		for (int j = 1; j <= n; j++)
		{
			cout << ans[i][j];
			if( j == n ) cout << '\n';
			else cout << ' ';
		}
	}
	return 0;
}