「中山纪中集训省选组D4T1」折射伤害 高斯消元

题目描述

在一个游戏中有n个英雄，初始时每个英雄受到数值为ai的伤害，每个英雄都有一个技能“折射”，即减少自己受到的伤害，并将这部分伤害分摊给其他人。对于每个折射关系，我们用数对\((x_i,y_i,z_i)\)来表示\(x_i\)将自己受到伤害去掉\(z_i\)的比例，将这些伤害转移给\(y_i\)（\(x_i,y_i\)是整数，\(z_i\)是实数）。

求出经过反复折射后最后每个英雄受到的实际总伤害。

输入格式

第一行一个正整数：\(n\)，表示有\(n\)个英雄，第二行\(n\)个整数\(A_i\)，依次表示每个英雄受到的初始伤害。第三行一个正整数\(m\)，表示有\(m\)对折射关系。接下来\(m\)行，每行三个数\(x_i,y_i,z_i\)，表示\(x_i\)将自己受到伤害去掉\(z_i\)的比例，将这些伤害转移给\(y_i\)。

输出格式

输出\(n\)行，第\(i\)行表示第\(i\)个英雄最后受到的实际总伤害。保留六位小数。

样例

样例输入

3
1 0 2
3
1 2 0.3000
1 2 0.2000
2 1 0.5000

样例输出

0.666667
0.333333
2.000000

数据范围

题解

设\(D_i\)为分摊伤害后\(i\)最终受到的伤害，\(C_i\)为\(i\)自己真实受到的的伤害占总伤害的比例，\(T_{i,j}\)表示\(i\)总共分给\(j\)的伤害的比例。

那么\(D_i\)中除了第一次天降的伤害\(A_i\)，其他伤害都是来源于别人的分摊，那么可以很轻松地推出一个式子：

\[\frac{D_i}{C_i}=\sum_{j≠i}\frac{D_j}{C_j}\cdot T_{j,i}+A_i
\]

直接按照这个式子建立关于\(D_i\)的方程组，然后高斯消元解方程即可。

\(Code:\)

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define N 205
#define M 20005
#define eps 1e-6
int n, m, A[N];
long double val[N][N], mat[N][N];
long double ans[N];
int main()
{
	scanf("%d", &n);
	for (int i = 1; i <= n; i++)
		scanf("%d", &A[i]), val[i][i] = 1;
	int x, y;
	double c;
	scanf("%d", &m);
	for (int i = 1; i <= m; i++)
	{
		scanf("%d%d%lf", &x, &y, &c);
		val[x][y] += c;
		val[x][x] -= c;
	}
	for (int i = 1; i <= n; i++)
	{
		for (int j = 1; j <= n; j++)
			if (i != j)
				mat[i][j] = -val[j][i] / val[j][j];
		mat[i][i] = 1 / val[i][i];
		mat[i][n + 1] = -A[i];
	}
	for (int i = 1; i <= n; i++)
	{
		for (int j = i + 1; j <= n; j++)
			if (fabs(mat[j][i]) > eps)
			{
				for (int k = i; k <= n + 1; k++)
					swap(mat[i][k], mat[j][k]);
				break;
			}
		long double s = mat[i][i];
		for (int j = i; j <= n + 1; j++)
			mat[i][j] /= s;
		for (int j = i + 1; j <= n; j++)
		{
			s = mat[j][i];
			for (int k = i; k <= n + 1; k++)
				mat[j][k] -= s * mat[i][k];
		}
	}
	ans[n + 1] = 1;
	for (int i = n; i >= 1; i--)
		for (int j = i + 1; j <= n + 1; j++)
			ans[i] -= mat[i][j] * ans[j];
	for (int i = 1; i <= n; i++)
		printf("%.6lf\n", double(ans[i]));
}