BZOJ4500: 矩阵

题意清晰明了简单易懂。
首先考虑这是一张二分图,设x[i]表示第i行增加的权值,y[i]表示第i列增加的权值,那么就有下面的一些限制关系:x[i]+y[i]==c。
于是我们将行当做左部点,列当做右部点,随便找一个初始点的权值设为随便一个数,dfs一遍判断是否矛盾即可…

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
//by:MirrorGray
using namespace std;
const int N=2111;
bool vis[N];
int a[N],tot=-1,head[N],ver[N],nxt[N],len[N];

void add(int x,int y,int z){
    nxt[++tot]=head[x];
    head[x]=tot;
    ver[tot]=y;
    len[tot]=z;
} 

void init(){
    tot=-1;
    memset(head,-1,sizeof(head));
    memset(vis,false,sizeof(vis));
}

bool dfs(int x){
    vis[x]=true;
    for(int i=head[x];~i;i=nxt[i])if(!vis[ver[i]]){
        a[ver[i]]=len[i]-a[x];
        if(!dfs(ver[i]))return false;
    }
    else if(a[ver[i]]+a[x]!=len[i])return false;
    return true;
}

int main(){
    int T;scanf("%d",&T);
    while(T--){
        init();
        int n,m,k;scanf("%d%d%d",&n,&m,&k);
        int pos=0;
        for(int i=1;i<=k;i++){
            int a,b,c;scanf("%d%d%d",&a,&b,&c);
            add(a,b+n,c);add(b+n,a,c);pos=a;
        }
        if(dfs(pos))puts("Yes");else puts("No");
    }
    return 0;
}
//orz Poker_face
//祈愿heoi2016一切平安…>_<…
### 题目内容 有一个球形空间产生器能够在n维空间中产生一个坚硬的球体。现在被困在了这个n维球体中,仅知道球面上n + 1个点的坐标,需要以最快的速度确定这个n维球体的球心坐标,以便于摧毁这个球形空间产生器[^1][^2]。 ### 解题思路 - **原理依据**:根据球心的定义,球心到球面上任意一点距离都相等。设球心坐标为\((x_1,x_2,\cdots,x_n)\),球面上一点坐标为\((a_1,a_2,\cdots,a_n)\),两点距离公式为\(dist = \sqrt{(x_1 - a_1)^2+(x_2 - a_2)^2+\cdots+(x_n - a_n)^2}\)。 - **构建方程**:设球面上\(n + 1\)个点的坐标分别为\((a_{i1},a_{i2},\cdots,a_{in})\),\(i = 1,2,\cdots,n + 1\)。以第一个点和其他点为例,根据球心到各点距离相等,可得: \((x_1 - a_{11})^2+(x_2 - a_{12})^2+\cdots+(x_n - a_{1n})^2=(x_1 - a_{21})^2+(x_2 - a_{22})^2+\cdots+(x_n - a_{2n})^2\) 展开式子: \(x_1^2 - 2a_{11}x_1+a_{11}^2+x_2^2 - 2a_{12}x_2+a_{12}^2+\cdots+x_n^2 - 2a_{1n}x_n+a_{1n}^2=x_1^2 - 2a_{21}x_1+a_{21}^2+x_2^2 - 2a_{22}x_2+a_{22}^2+\cdots+x_n^2 - 2a_{2n}x_n+a_{2n}^2\) 消去\(x_1^2,x_2^2,\cdots,x_n^2\)后可得: \(2(a_{21}-a_{11})x_1 + 2(a_{22}-a_{12})x_2+\cdots+2(a_{2n}-a_{1n})x_n=a_{21}^2 - a_{11}^2+a_{22}^2 - a_{12}^2+\cdots+a_{2n}^2 - a_{1n}^2\) 同理,用第一个点和第\(i\)个点\((i = 3,\cdots,n + 1)\)可得到\(n\)个线性方程,构成一个\(n\)元一次方程组。 - **求解方程组**:使用高斯消元法求解这个\(n\)元一次方程组,得到的解就是球心的\(n\)维坐标。 ### 代码实现 ```python n = int(input()) points = [] for _ in range(n + 1): points.append(list(map(float, input().split()))) # 构建方程组的系数矩阵和常数项 a = [[0] * (n + 1) for _ in range(n)] b = [0] * n for i in range(n): for j in range(n): a[i][j] = 2 * (points[i + 1][j] - points[0][j]) b[i] += points[i + 1][j] ** 2 - points[0][j] ** 2 # 高斯消元 for i in range(n): # 选主元 max_row = i for j in range(i + 1, n): if abs(a[j][i]) > abs(a[max_row][i]): max_row = j a[i], a[max_row] = a[max_row], a[i] b[i], b[max_row] = b[max_row], b[i] # 消元 for j in range(i + 1, n): factor = a[j][i] / a[i][i] for k in range(i, n): a[j][k] -= factor * a[i][k] b[j] -= factor * b[i] # 回代求解 x = [0] * n for i in range(n - 1, -1, -1): s = 0 for j in range(i + 1, n): s += a[i][j] * x[j] x[i] = (b[i] - s) / a[i][i] # 输出结果 print(" ".join(map(lambda num: "{:.3f}".format(num), x))) ```
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值