HPU6004——Happy path dfs思想 树形dp的应用

题意就是给定一个有向无环图，问有多少个不同的路径；

题目链接：http://net.hpuacm.cf/problem/6004

可以看出因为是有向无环图所以就可以看成树的形式，然后就是考虑如何计算路径数，对于路径数，可以考虑解决子结构，把一个子结构解决了，也就解决了整个树。

我们可以计算每一个子根节点下的路径数，然后把所有的子根节点累加起来就是结果；

代码：

#include<bits/stdc++.h>

using namespace std;

const int maxn = 1e5 + 10;
const int mod = 1e9 + 7;

int in[maxn];	//记录出入度，要从入度为0的点开始搜索
int vis[maxn];//
int dp[maxn];//用来记录每一个节点有多少条路径
int son[maxn];//用来标记每一个节点下有多少子节点；
int n, m;
vector<int>oj[maxn];

void dfs(int x)
{
	if (oj[x].size() == 0)
	{
		son[x] = 1;
		return;
	}
	for (int i = 0; i < oj[x].size(); i++)
	{
		int t = oj[x][i];
		if (vis[t])
		{
			son[x] += son[t];//这里不再让dp【x】加是因为在这个点被标记时他的下面那些单线路已经被计算了
			son[x] %= mod;
			continue;
		}
		vis[t] = 1;
		dfs(t);
		dp[x] += dp[t];
		son[x] += son[t];
		dp[x] %= mod, son[x] %= mod;
	}
	dp[x] += son[x]++;//这里就是记录三个点的路径个数
	dp[x] %= mod;
}

void ini()
{
	for (int i = 1; i < maxn; i++)
	{
		oj[i].clear();
	}
	memset(vis, 0, sizeof(vis));
	memset(in, 0, sizeof(in));
	memset(dp, 0, sizeof(dp)); 
	memset(son, 0, sizeof(son));
}

int main()
{
	int t;
	scanf("%d", &t);
	while (t--)
	{
		scanf("%d %d", &n, &m);
		ini();
		for (int i = 1; i <= m; i++)
		{
			int a, b;
			scanf("%d %d", &a, &b);
			in[b]++;
			oj[a].push_back(b);
		}
		for (int i = 1; i <= n; i++)
		{
			sort(oj[i].begin(), oj[i].end());
			oj[i].erase(unique(oj[i].begin(), oj[i].end()), oj[i].end());
		}
		long long ans = 0;
		for (int i = 1; i <= n; i++)
		{
			if (!in[i])
			{
				dfs(i);
				ans += dp[i];
				ans %= mod;
			}
		}
		cout << ans << endl;
	}
	return 0;
}