1495D BFS Trees

一道有趣的题目。

因为n和m都很小，所以应该是枚举两个点来求方案数。

我们先要bfs两次把每个点离i，j的的距离求出来，然后我们可以发现，对于一个点。

然后利用转化思想，把两个点变成x和y坐标放到平面直角坐标系上面去。

然后自己手玩一下最普遍的生成树，i和j之间有一段路径，这个路径的每个点都有一写子树。把这个图弄到坐标系上面去就是（个数）：

i处在5的位置，j处在E的位置，那一段1是他们之间的路径，然后？表示接在路径上面的点（因为个数不确定）。

然后我们可以发现，路径上面只能是1，因此要特判这种情况。又很容易发现会有一些点不能连到路径上面去，那么答案就是0。

特判完答案是0的情况以后，只要求一下结果就行了。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=410,M=610,mod=998244353;
int n,m,p[2][N],cnt[N][N];
int tot,hd[N],ver[M<<1],nxt[M<<1];
void add(int x,int y){
	tot++;
	ver[tot]=y;
	nxt[tot]=hd[x];
	hd[x]=tot;
}
void bfs(int w,int st){
	queue<int>q;
	memset(p[w],-1,sizeof(p[w]));p[w][st]=0;
	q.push(st);
	while(q.size()){
		int x=q.front();q.pop();
		for(int i=hd[x];i;i=nxt[i]){
			int y=ver[i];
			if(p[w][y]!=-1)continue;
			p[w][y]=p[w][x]+1;
			q.push(y);
		}
	}
}
int ax(int k){return p[0][k];}
int ay(int k){return p[1][k];}
bool pd(int k,int i,int j){
	if(abs(ax(k)+ay(k)-ay(i))&1)return 0;
	int tx=ax(k),ty=ay(k);
	int cc=abs(ax(k)+ay(k)-ay(i))/2;
	tx-=cc,ty-=cc;
	if(tx<0||ty<0)return 0;
	return 1;
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=m;i++){
		int x,y;scanf("%d%d",&x,&y);
		add(x,y),add(y,x);
	}
	for(int i=1;i<=n;i++){
		bfs(0,i);
		for(int j=1;j<=n;j++){
			bfs(1,j);
			for(int k=1;k<=n;k++)
				if(k!=i&&k!=j)cnt[ax(k)][ay(k)]++;
			bool bk=1;
			for(int k=1;k<=n;k++)
				if(k!=i&&k!=j&&!pd(k,i,j)){
					bk=0;
					break;
				}
			int tx=1,ty=ay(i)-1;
			while(ty>0){
				if(cnt[tx][ty]!=1){
					bk=0;
					break;
				}
				tx++,ty--;
			}
			if(bk){
				int ans=1;
				for(int k=1;k<=n;k++)
					if(k!=i&&k!=j&&ax(k)+ay(k)!=ay(i)){
						int cnt=0;
						for(int ii=hd[k];ii;ii=nxt[ii]){
							int y=ver[ii];
							if(ax(y)==ax(k)-1&&ay(y)==ay(k)-1)cnt++;
						}
						ans=1ll*ans*cnt%mod;
					}
				printf("%d ",ans);
			}
			else printf("0 ");
			for(int k=1;k<=n;k++)
				if(k!=i&&k!=j)cnt[ax(k)][ay(k)]--;
		}
		puts("");
	}
	return 0;
}