hdu 3976 Electric resistance

高斯消元的模板题

注意用电路中的KCL定理

#include <CSTDIO>
#include <QUEUE>
using namespace std;

/*
hdu 3976
高斯消元
每个节点的流入电流等于流出电流
R = 总电势差/电流
列出方程：
u[i]代表节点i的电势
Σ((u[j]-u[i])/r[i][j]) + I = 0
I[0] = 1
I[n-1] = -1
*/
const int MAXN = 55;
const double _inf = 1e-9;
double a[MAXN][MAXN], x[MAXN]; // 方程左边的矩阵和等式右边的值， x存放最后结果
int equ, val;	// 方程数 未知数个数
inline double mabs(double _X){return _X<0?-_X:_X;}
int gauss()
{
	int i,j,k,col,max_r;
    for(k=0,col=0;k<equ&&col<val;k++,col++)
    {
        max_r=k;
        for(i=k+1;i<equ;i++)
		{
			if(mabs(a[i][col])>mabs(a[max_r][col]))
				max_r=i;
		}
		if(mabs(a[max_r][col])<_inf) return 0;
		if(k!=max_r)
		{
			for(j=col;j<val;j++)
				swap(a[k][j],a[max_r][j]);
			swap(x[k],x[max_r]);
		}
		x[k]/=a[k][col];
		for(j=col+1;j<val;j++)a[k][j]/=a[k][col];
		a[k][col]=1;
		for(i=0;i<equ;i++)
		{
			if(i!=k)
			{
				x[i]-=x[k]*a[i][k];
				for(j=col+1;j<val;j++)a[i][j]-=a[k][j]*a[i][col];
				a[i][col]=0;
			}
		}
    }
    return 1;
}

int n, m;
double data[MAXN][MAXN];
int main()   
{
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
#endif
	int t, cs = 0;
	int i, j;
	double tmp;
	scanf("%d", &t);
	while (t--)
	{
		scanf("%d%d", &n, &m);
		memset(data, 0, sizeof data);
		while (m--)
		{
			scanf("%d%d%lf", &i, &j, &tmp);
			--i, --j;
			data[i][j] = data[j][i] = tmp;
		}
		memset(a, 0, sizeof a);
		memset(x, 0, sizeof x);
		for (i = 0; i< n; ++i)
		{
			for (j = 0; j< n; ++j)
			{
				if (data[i][j])
				{
					a[i][j] += 1.0/data[i][j];
					a[i][i] -= 1.0/data[i][j];
				}
			}
		}
		x[0] = -1;
		x[n-1] = 1;
		equ = val = n;
		gauss();
		printf("Case #%d: %.2lf\n", ++cs, x[0]-x[n-1]);
	}
	return 0;
}