Happy Path（dfs）

最新推荐文章于 2023-10-17 23:53:23 发布

我要去看得最远的地方

最新推荐文章于 2023-10-17 23:53:23 发布

阅读量1k

点赞数

分类专栏： HPU第6次积分赛

本文链接：https://blog.csdn.net/qq_42434171/article/details/82222804

版权

HPU第6次积分赛专栏收录该内容

6 篇文章 0 订阅

订阅专栏

题解：现在总结一下unique，unique的作用是“去掉”容器中相邻元素的重复元素（不一定要求数组有序），它会把重复的元素添加到容器末尾（所以数组大小并没有改变），而返回值是去重之后的尾地址，

vector函数的erase函数erase函数可以用于删除vector容器中的一个或者一段元素，在删除。第一个参数为删除的首地址，第二个参数为删除的末位置。

Input

Output

Sample Input 1

Sample Output 1

#include <cstdio>
#include <algorithm>
#include <vector>
#include <cmath>
#include <cstring>
using namespace std;
const int MAXN = 200010;
const int mod = 1e9 + 7;
vector<int> G[MAXN];
int vis[MAXN], in[MAXN];
int son[MAXN], dp[MAXN];
long long ans;

void dfs(int x) {
	if(!G[x].size()) {//所有点的根节点G[x].size()=0 
		son[x] = 1;
		return ;
	}
	for(int i = 0; i < G[x].size(); ++i) {
		int u = G[x][i];
		if(vis[u]) {
			son[x] += son[u];
			son[x] %= mod;
			continue;
		}
		vis[u] = 1;
		dfs(u);
		dp[x] += dp[u];//第一次出现加上u 到根节点的路径数 
		son[x] += son[u];
		dp[x] %= mod, son[x] %= mod;
	}
	dp[x] += son[x]++;
	dp[x] %= mod, son[x] %= mod;
}

int main() {
	 // freopen("in3.txt", "r", stdin);
	 // freopen("out3.txt", "w", stdout);
	int n, m, T;
	scanf("%d", &T);
	while(T--) {
		for(int i = 0; i < MAXN; ++i) G[i].clear(), son[i] = 0;
		for(int i = 0; i < MAXN; ++i) vis[i] = in[i] = dp[i] = 0;
		scanf("%d %d", &n, &m);
		for(int i = 0; i < m; ++i) {
			int a, b;
			scanf("%d %d", &a, &b);
			in[b]++;//直接指向b节点的个数 
			G[a].push_back(b);
		}
		for(int i = 1; i <= n; ++i) {
            sort(G[i].begin(), G[i].end());
            G[i].erase(unique(G[i].begin(), G[i].end()), G[i].end());//删除重边 
        }
		ans = 0;
		for(int i = 1; i <= n; ++i) {
			if(in[i]) continue;//寻找子节点 
			dfs(i);
			ans += dp[i];
			ans %= mod;
		}
		printf("%lld\n", ans);
	}
	return 0;
}