B 有向无环图

题目描述

Bobo 有一个 n 个点,m 条边的有向无环图(即对于任意点 v,不存在从点 v 开始、点 v 结束的路径)。
为了方便,点用 1,2,…,n1, 2, \dots, n1,2,…,n 编号。
设 count(x,y)\mathrm{count}(x, y)count(x,y) 表示点 x 到点 y 不同的路径数量(规定 count(x,x)=0\mathrm{count}(x, x) = 0count(x,x)=0),Bobo 想知道
∑i=1n∑j=1ncount(i,j)⋅ai⋅bj\sum_{i = 1}^n\sum_{j = 1}^n \mathrm{count}(i, j) \cdot a_i \cdot b_j∑i=1n​∑j=1n​count(i,j)⋅ai​⋅bj​
除以 (109+7)(10^9+7)(109+7) 的余数。
其中,ai,bja_i, b_jai​,bj​ 是给定的数列。

输入描述:

输入包含不超过 15 组数据。
每组数据的第一行包含两个整数 n, m (1≤n,m≤1051 \leq n, m \leq 10^51≤n,m≤105).
接下来 n 行的第 i 行包含两个整数 ai,bia_i, b_iai​,bi​ (0≤ai,bi≤1090 \leq a_i, b_i \leq 10^90≤ai​,bi​≤109).
最后 m 行的第 i 行包含两个整数 ui,viu_i, v_iui​,vi​,代表一条从点 uiu_iui​ 到 viv_ivi​ 的边 (1≤ui,vi≤n1 \leq u_i, v_i \leq n1≤ui​,vi​≤n)。

输出描述:

对于每组数据,输出一个整数表示要求的值。

示例1
输入
复制

3 3
1 1
1 1
1 1
1 2
1 3
2 3

输出
复制

4

示例2
输入
复制

2 2
1 0
0 2
1 2
1 2

输出
复制

4

示例3
输入
复制

2 1
500000000 0
0 500000000
1 2

输出
复制

250000014

找u到v的路径数乘以a[i] , b[j],可以先找du为零的点所直接能到达的边,ans加上a[i] * b[j], 让a[j]的值加上a[i]的值,然后du–, 执行n次,所得的结果是一样的。
最后,不要忘了取模,相加时取模,相乘时取模。
本来以为相加时不会溢出,就没有取模,导致一直wa,,还是太天真了。。

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#define mem(a, b) memset(a, b, sizeof(a))
using namespace std;

const int maxn = 1e5 + 10;
const int mod = 1e9 + 7;
typedef long long ll;
struct node{
	int v, nex;
	node(int _v = 0, int _nex = 0)
	{
		v = _v, nex = _nex;
	}
}edge[maxn];
int head[maxn], du[maxn];
int n, m, cnt = 1;
int a[maxn], b[maxn];
ll ans;

void addedge(int u, int v)
{
	edge[cnt].v = v;
	edge[cnt].nex = head[u];
	head[u] = cnt++;
}

void init()
{
	mem(head, -1);
	mem(a, 0);
	mem(b, 0);
	mem(du, 0), mem(edge, 0);
	cnt = 1;
	ans = 0;
}

void solve(int x)
{
	for(int i = head[x]; i != -1; i = edge[i].nex)
	{
		int v = edge[i].v;
		du[v]--;
		(a[v] += a[x]) %= mod;
		(ans += 1ll * a[x] * b[v] % mod) %= mod;
	}
	du[x] = -1;
}

int main()
{
	while(scanf("%d%d", &n, &m) != EOF)
	{
		init();
		int x, y;
		for(int i = 1; i <= n; i++)
			scanf("%d%d", &a[i], &b[i]);
		for(int i = 1; i <= m; i++)
		{
			scanf("%d%d", &x, &y);
			addedge(x, y);
			du[y]++;
		}
		int num = 0;
		int f = 1;
		while(num < n)
		{
			f = 0;
			for(int i = 1; i <= n; i++)
			{
				if(du[i] == 0)
				{
					solve(i);
					num++;
				}
			}
		}
		printf("%lld\n", ans);
	}
	return 0;
}
发布了73 篇原创文章 · 获赞 15 · 访问量 8107

猜你喜欢

转载自blog.csdn.net/ln2037/article/details/102335210
今日推荐