装备购买[BZOJ4004]

欢迎大家访问我的老师的OJ———caioj.cn

题面描述

传送门

思路

提要:

做此题时，请务必了解线性空间的基以及矩阵的秩。

把$n$件装备看作$n$个长度为$m$的向量，

根据题意，购买的装备对应的向量应该是线性无关的(要满足向量不能被其他向量表出)。

要买下最多数量的装备，其实就是求线性空间的任意一个基的个数。

把$a_{i.j}(1\le i\le n,1\le j\le m)$看作系数矩阵，每个装备$z_i$都是一个行向量。

用高斯消元求出该矩阵的秩(相当于线性空间的基中元素个数)，就得到了能买下的装备的最多数量。

本题还需要求花最少的钱。我们只需要在高斯消元的过程中，使用贪心策略，对于每一个主元$x_i$，在前$i-1$列为$0$、第$i$列不为$0$的行向量中，选择价格最低的一个，小区其他行中i列的值。

用反证法证明。假设花费价钱最少的基底(简称基)$z[i_1],z[i_2],z[i_3],......,z[i_p]$不包含价格最少的行向量$z[k]$，因为基底是极大线性无关子集，所以$z[k]$能被$z[i_1],z[i_2],z[i_3],......,z[i_p]$表出，不妨设$z[k]$=$b_{1}z[i_1]+b_{2}z[i_2]+b_{3}z[i_3]+......+b_{p}z[i_p]$.

移向就有$z[i_p]=(z[k]-b_{1}z[i_1]-b_{2}z[i_2]-b_{3}z[i_3]-......-b_{p-1}z[i_{p-1}])/b_p$，即$z[i_p]$能被$z[k]$与$z[i_1],z[i_2],z[i_3],......,z[i_{p-1}]$表出。

所以这两个基都能表出相同的线性空间，而含有$z[k]$明显总费用最低，与假设矛盾。

证毕。

AC code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#define eps 1e-6
using namespace std;
long double a[510][510];
int n,ans,m,dim,c[510];
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
		{
			double tmp;scanf("%lf",&tmp);a[i][j]=tmp;
		}
	for(int i=1;i<=n;i++)scanf("%d",&c[i]);
	dim=0;ans=0;
	for(int i=1;i<=m;i++)
	{
		int x=0;
		for(int j=dim+1;j<=n;j++)
			if(fabs(a[j][i])>eps&&(x==0||c[x]>c[j]))x=j;
		if(x==0)continue;
		ans+=c[x];dim++;
		for(int j=dim;j<=m;j++)swap(a[dim][j],a[x][j]);
		swap(c[x],c[dim]);
		for(int k=dim+1;k<=n;k++)
		{
			if(fabs(a[k][i])>eps)
			{
				for(int j=m;j>=i;j--)a[k][j]-=a[k][i]/a[dim][i]*a[dim][j];
			}
		}
	}	
	printf("%d %d\n",dim,ans);
	return 0;
}