BZOJ 2419: 电阻 高斯消元

2419: 电阻

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 358  Solved: 153
[Submit][Status][Discuss]

Description

你突破了无数艰难险阻，终于解决了上面那道题，众神犇瞬间就
震惊了。他们发现居然有人可以把那种非人类做的题目做出来。
他们一致同意，最后这道题不能再出数学题了。考虑到两位小盆
友的状态，他们决定考考你的初中物理水平。 
一个电路板，有 N 个接点，M 个电阻。电阻两端都在接点上，
都告诉了你阻值。询问1 号点与N 号点的等效电阻是多少。 
两位小盆友很聪明，他们拿出了一个欧姆表，瞬间就虐爆了…不
过你也不甘落后，你肯定不会用这种投机取巧、误差巨大的方法，
于是你看向了你的电脑。

Input

多组数据，输入直到文件结束 
每组数据第一行两个整数N，M 
接下来 M 行，每行三个非负整数 X，Y，R，表示电阻连接的两
接点和阻值。

Output

每组数据输出一行，一个实数，四舍五入到小数点后两位 
 

Sample Input

2 1
1 2 1

Sample Output

1.00

太强了：http://blog.csdn.net/jiangshibiao/article/details/28661223

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<set>
#include<map>
using namespace std;

typedef double db;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=510;

int n,m;

db a[N][N],f[N][N],ans[N];

void gauss()
{
	register int i,j,k;
	db tmp;
	for(i=1;i<=n;++i)
	{
		for(k=0,j=i;j<=n;++j)
			if(fabs(f[j][i])>fabs(f[k][i]))
				k=j;
		for(j=1;j<=n+1;++j) swap(f[i][j],f[k][j]);
		for(j=i+1;j<=n;++j)
		{
			tmp=f[j][i]/f[i][i];
			for(k=i;k<=n+1;++k)
				f[j][k]=f[j][k]-f[i][k]*tmp;
		}
	}
	for(i=n;i;i--)
	{
		for(j=n;j>i;j--)
			f[i][n+1]-=ans[j]*f[i][j];
		ans[i]=f[i][n+1]/f[i][i];
	}
}

int main()
{
	register int i,j,x,y,R;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		memset(a,0,sizeof(a));memset(f,0,sizeof(f));
		for(i=1;i<=m;++i)
			x=read(),y=read(),R=read(),
			a[x][y]+=1.0/R,a[y][x]+=1.0/R;
		for(i=1;i<=n;++i)
			for(j=1;j<=n;++j)
				f[i][i]+=a[i][j],
				f[i][j]-=a[i][j];
		f[1][n+1]=1;f[n][n+1]=-1;f[n][n]+=1.0;
		gauss();
		printf("%.2lf\n",ans[1]);
	}
	return 0;
}