每日一题之hiho1195 （高斯消元模版）

描述
小Ho：喂不得了啦，那边便利店的薯片半价了!

小Hi：啥?!

小Ho：那边的便利店在打折促销啊。

小Hi：走走走，赶紧去看看=v=

于是小Hi和小Ho来到了便利店。

老板为了促销，推出了组合包的形式，将不同数量的各类商品打包成一个组合，顾客可以选择组合进行购买。比如2袋薯片，1听可乐的组合只要5元，而1袋薯片，2听可乐的组合只要4元。

通过询问老板，小Hi和小Ho知道：一共有N种不同的商品和M种不同的商品组合；每一个组合的价格等于组合内商品售价之和，一个组合内同一件商品不会超过10件。

小Hi：这样算下来的话，一听可乐就是1元，而一包薯片是2元。小Ho，如果你知道所有的组合情况，你能分别算出每一件商品单独的价格么？

小Ho：当然可以了，这样的小问题怎么能难到我呢？

提示：高斯消元

输入
第1行：2个正整数，N,M。表示商品的数量N，组合的数量M。1≤N≤500, N≤M≤2*N

第2..M+1行：N+1个非负整数，第i+1行第j列表示在第i个组合中，商品j的数量a[i][j]。第i+1行第N+1个数表示该组合的售价c[i]。0≤a[i][j]≤10, 0≤c[i]≤10^9

输出
若没有办法计算出每个商品单独的价格，输出”No solutions”

若可能存在多个不同的结果，输出”Many solutions”

若存在唯一可能的结果，输出N行，每行一个非负整数，第i行表示第i个商品单独的售价。数据保证如果存在唯一解，那么解一定恰好是非负整数解。

样例输入
2 2
2 1 5
1 2 4
样例输出
2
1

#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>

using namespace std;

const int MAX_N = 505;
const int MAX_M = 1005;
const double EPS = 1e-7;

struct Answer {
  double x;
  int id;
};

Answer ans[MAX_N];
double a[MAX_M][MAX_N];
double b[MAX_M];
double tmp[MAX_N];

void swap_line(int i, int j) {
  memcpy(tmp, a[i], sizeof(tmp));
  memcpy(a[j], a[i], sizeof(tmp));
  memcpy(a[i], tmp, sizeof(tmp));
  swap(b[i], b[j]);
  swap(ans[i], ans[j]);
}

int guass_slove(int n, int m) {
  // 处理出上三角矩阵
  for (int i = 1; i <= n; i++) {
    bool flag = false;
    for (int j = i; j <= m; j++) {  // 从第i行开始，找到第i列不等于0的行j
      if (fabs(a[j][i]) > EPS) {
        swap_line(j, i);            // 交换第i行和第j行
        flag = true;
        break;
      }
    }
    // 若无法找到，则存在多个解
    if (!flag)
      return 2;                     // 解多于一个
    // 消除第i+1行到第M行的第i列
    for (int j = i + 1; j <= m; j++)
      tmp[j] = a[j][i] / a[i][i];
    for (int j = i + 1; j <= m; j++) {
      for (int k = 1; k <= n; k++) {
        a[j][k] = a[j][k] - a[i][k] * tmp[j];
      }
      b[j] = b[j] - b[i] * tmp[j];
    }
  }
  // 检查是否无解，即存在 0 = x 的情况
  for (int i = 1; i <= m; i++) {
    if (fabs(b[i]) < EPS)
      continue;
    bool all_zero = true;           // 第i行系数均为0?
    for (int j = 1; j <= n; j++) {
      if (fabs(a[i][j]) > EPS) {
        all_zero = false;
        break;
      }
    }
    if (all_zero)
      return 0;                     // 无解
  }
  // 此时存在唯一解
  // 由于每一行都比前一行少一个系数，所以在M行中只有前N行有系数
  // 从第N行开始处理每一行的解
  for (int i = n; i >= 1; i--) {
    // 利用已经计算出的结果，将第i行中第i+1列至第N列的系数消除
    for (int j = i + 1; j <= n; j++) {
      b[i] = b[i] - a[i][j] * ans[j].x;
      a[i][j] = 0;
    }
    ans[i].x = b[i] / a[i][i];
  }
  return 1;                        // 唯一解
}

bool id_order(const Answer& a, const Answer& b) {
  return a.id < b.id;
}

int main() {
  int n, m;
  scanf("%d %d", &n, &m);
  for (int i = 1; i <= m; i++) {
    for (int j = 1; j <= n; j++)
      cout<< a[i][j] << endl;
    cout << b[i] << endl;
    ans[i].id = i;
  }
  int res = guass_slove(n, m);
  if (res == 0)
    cout << "No solutions" << endl;
  else if (res == 2)
    cout << "Many solutions" << endl;
  else {
    sort(ans + 1, ans + n + 1, id_order);
    for (int i = 1; i <= n; i++)
      cout << int(ans[i].x + 0.5) << endl;
  }
  return 0;
}