题解:SP2415 RESIST - Kirchhof Law
概述给定 n 点 m 边的无向图,边权为电阻阻值,求节点 1 到 n 的等效电阻。设各节点电势,由欧姆定律得每条边的电流,再对每个普通节点用基尔霍夫电流定律令支路电流代数和为 0,每条边给系数矩阵对应的对角与非对角元贡献电导,组成关于各点电势的线性方程组。在节点 1 注入 1 安培电流、把节点 n 设为零电势,高斯消元解出节点 1 的电势即为等效电阻。时间复杂度 O(n 的 3 次方)。
给定一张 n 个节点 m 条边的无向图,边权表示电阻阻值,求节点 1 到 n 的等效电阻。
设节点 x 的电势为 φx,边 (u,v) 的电阻为 Ru,v,则电导为:
Gu,v=Ru,v1
根据 欧姆定律,从 u 到 v 的电流为:
Iu,v=Ru,vφu−φv=Gu,v(φu−φv)
根据 基尔霍夫电流定律,对于任意一个普通节点,所有关联支路电流的代数和为 0。
考虑节点 u 的电流:
(u,v)∈E∑Iu,v=(u,v)∈E∑Gu,v(φu−φv)=0
展开可得:
(u,v)∈E∑Gu,vφu−(u,v)∈E∑Gu,vφv=0
这是一个关于各点电势的线性方程。
au,1φ1+au,2φ2+⋯+au,nφn=bu
考虑添加一条边 (u,v),设其电导为 G。节点 u 的方程加入 G(φu−φv),节点 v 的方程加入 G(φv−φu),因此对应到系数矩阵就是:
au,u←au,u+G
au,v←au,v−G
av,v←av,v+G
av,u←av,u−G
把所有边的贡献都加到矩阵里,就得到了整个线性方程组。
我们可以在节点 1 通入 1A 电流,并将节点 n 设为零电势点,求出节点 1 的电势。
b1=1,φn=0
利用 高斯消元 求解各点电势后,等效电阻为:
R=IU=1φ1−φn=φ1
#include <bits/stdc++.h>
using namespace std;
const double eps=1e-8;
const int N=105;
double a[N][N];
bool gauss(int n)
{
for(int i=1;i<=n;i++)
{
int t=i;
for(int j=i+1;j<=n;j++)
{
if(fabs(a[j][i])>fabs(a[t][i]))t=j;
}
if(fabs(a[t][i])<eps)return 0;
swap(a[i],a[t]);
for(int j=n+1;j>=i;j--)a[i][j]/=a[i][i];
for(int j=1;j<=n;j++)
{
if(i==j)continue;
for(int k=n+1;k>i;k--)a[j][k]-=a[i][k]*a[j][i];
}
}
return 1;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n,m;
while(cin>>n>>m)
{
while(m--)
{
int u,v;
double r;
cin>>u>>v>>r;
double g=1/r;
a[u][u]+=g;
a[u][v]-=g;
a[v][v]+=g;
a[v][u]-=g;
}
a[1][n+1]=1;
for(int i=1;i<n;i++)a[n][i]=0;
a[n][n]=1;
a[n][n+1]=0;
gauss(n);
cout<<fixed<<setprecision(2)<<a[1][n+1]<<'\n';
memset(a,0,sizeof a);
}
return 0;
}