洛谷 [P3265] 装备购买

线性基

通过题目描述可以感觉到就是要求线性基,
线性基的求法是高斯消元,消完以后剩下的x的系数非 0 的就是线性基
本题有一个贪心策略,每次挑选价格最小的来消掉其他的元

//可以快排预处理
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
const int MAXN = 505;
int b[MAXN], n, m;
long double a[MAXN][MAXN];
int main() {
	cin >> n >> m;
	for(int i = 1; i <= n; i++) {
		for(int j = 1; j <= m; j++) {
			cin >> a[i][j];
		}
	}
	for(int i = 1; i <= n; i++) cin >> b[i];
	for(int i = 1; i <= m; i++) {
		for(int j = i + 1; j <= n; j++) {
			if(fabs(a[j][i]) > 1e-4 && b[j] < b[i]) {
				for(int k = 1; k <= n; k++) swap(a[i][k], a[j][k]);
				swap(b[i], b[j]);
			}
		}
		if(!a[i][i]) continue;
		for(int j = 1; j <= n; j++) {
			if(i == j) continue;
			long double rate = (a[j][i] / a[i][i]);
			for(int k = i; k <= m; k++) {
				a[j][k] -= a[i][k] * rate;
			}
		}
	}
	int cnt = 0, ans = 0;
	for(int i = 1; i <= min(n, m); i++) {
		if(fabs(a[i][i]) > 1e-4) cnt++, ans += b[i];
	}
	printf("%d %d\n", cnt, ans);
	return 0;
}

线性基优化版

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#define eps 1e-5
using namespace std;
const int MAXN = 505;
int n, m, nxt[MAXN], cnt, ans;
struct qwq{
	long double num[MAXN];
	int val;
	bool operator < (const qwq &b) const{
		return this -> val < b.val;
	}
}a[MAXN];
bool f[MAXN];
int main() {
	cin >> n >> m;
	for(int i = 1; i <= n; i++) {
		for(int j = 1; j <= m; j++) {
			cin >> a[i].num[j];
		}
	}
	for(int i = 1; i <= n; i++) cin >> a[i].val;
	sort(a + 1, a + n + 1);
	for(int i = 1; i <= n; i++) {
		for(int j = 1; j <= m; j++) {
			if(fabs(a[i].num[j]) > eps){
				if(!f[j]) {
					f[j] = 1;
					cnt++;
					ans += a[i].val;
					nxt[j] = i;
					break;
				}else{
					long double rate = a[i].num[j] / a[nxt[j]].num[j];
					for(int k = j; k <= n; k++) a[i].num[k] -= rate * a[nxt[j]].num[k];
				}
			}	
		}
	}
	printf("%d %d\n", cnt, ans);
	return 0;
}