树形图计数

树形图计数

【问题描述】

小k同学最近正在研究最小树形图问题。所谓树形图，是指有向图的一棵有根的生成树，其中树的每一条边的指向恰好都是从根指向叶结点的方向。现在小k在纸上画了一个图，他想让你帮忙数一下这个图有多少棵树形图。

【输入格式】

第1行输入1个正整数：n，表示图中点的个数

第2~n+1行每行输入n个字符，描述了这个图的邻接矩阵。第n+1行第n个字符如果是0则表示没有从n连向n+1的有向边，1表示有一条从n到n+1的有向边。

【输出格式】输出1行1个整数，表示这个有向图的树形图个数。

【输入样例】

4

0100

0010

0001

1000

【输出样例】4

【数据规模和约定】

思路：以每一个节点为根，判断从这个根开始用有向边能不能找到所有节点，如果能则记录。

对于100%的数据，n≤8.

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
int n, ans,g[1000][1000];
int f[1000];
char st[1000];

int find(int x) {
	if (f[x]!=x) f[x]=find(f[x]);  
    return f[x];  
}
//用并查集判断是否成环（生成树)
void dfs(int root, int v) {
    if (v == root) dfs(root, v + 1);
    if (v == n) {ans++;return;}
    for (int k=0;k<n;k++)
        if (k != v && g[k][v] == 1 && find(k) != v) {
            f[v] = k;
            dfs(root, v + 1);
            f[v] = v;
        }
}

int main() {
    scanf("%d", &n);
	for (int i=1;i<=n;i++) f[i]=i;
   	for (int i=0;i<n;i++){
		scanf("%s",st);
		for (int j=0;j<n;j++) g[i][j]=st[j]-48;
	}
	ans = 0;
    for (int k = 0; k < n; ++ k)
        dfs(k, 0);      
    printf("%d\n", ans);   
    return 0;
}