hdu3359（浮点数高斯消元）

题意：H * W (W,H <= 10) 的矩阵A的某个元素A[i][j]，从它出发到其他点的曼哈顿距离小于等于D的所有值的和S[i][j]除上可达点的数目，构成了矩阵B。给定矩阵B，求矩阵A。

思路：根据题意直接构造增广矩阵，然后用高斯消元直接做，最后得到一维的所有结果，浮点数高斯消元入门题

#include<set>
#include<queue>
#include<cmath>
#include<cstdio>
#include<vector>
#include<string>
#include<utility>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
#define Inf (1<<30)
#define LL long long
#define MOD 1000000009
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
int n,m,d,cnt;
double a[20][20];
double resx[210];
double dp[210][210];
const double eps=1e-12;
double Count(int x,int y)
{
    double ans=0;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
            if(abs(i-x)+abs(j-y)<=d)ans++;
    }
    return ans;
}
void init()
{
    int num=0;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            double down=Count(i,j);
            dp[num][i*m+j]=1;
            for(int k=0;k<n;k++)
            {
                for(int z=0;z<m;z++)
                {
                    if(abs(k-i)+abs(z-j)<=d)
                        dp[num][k*m+z]=1.0/down;
                }
            }
            dp[num++][cnt]=a[i][j];
        }
    }
}
void gauss()
{
    for(int r=0;r<cnt;r++)
    {
        int max_r=r;
        for(int i=r+1;i<cnt;i++)
        {
            if(fabs(dp[i][r])>fabs(dp[max_r][r]))max_r=i;///找到最大的一行
        }
        if(max_r!=r)
        {
            for(int i=0;i<cnt+1;i++)swap(dp[r][i],dp[max_r][i]);///当前行与最大的一行交换
        }
        if(fabs(dp[r][r])<eps) continue;
        for(int k=r+1;k<cnt;k++)///对角化
        {
            double ta=dp[k][r]/dp[r][r];
            for(int j=r;j<cnt+1;j++)
                dp[k][j]-=ta*dp[r][j];
        }
    }
    for(int i=cnt-1;i>=0;i--)///回代求解
    {
        double temp=dp[i][cnt];
        for(int j=i+1;j<cnt;j++)
            temp-=dp[i][j]*resx[j];
        resx[i]=temp/dp[i][i];
    }
}
int main() {
    int T=0;
    //freopen("D:\\oo.txt","r",stdin);
    while(~scanf("%d%d%d",&m,&n,&d)&&n+m+d)
    {
        cnt=n*m;
        if(T)puts("");
        memset(dp,0,sizeof(dp));
        for(int i=0;i<n;i++)for(int j=0;j<m;j++)scanf("%lf",&a[i][j]);
        init();gauss();
        int num=0;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)printf("%8.2lf",resx[num++]+eps);
            puts("");
        }
        T++;
    }
    return 0;
}