洛谷 P5027 Barracuda 【高斯消元】

题目链接

题意：

给出 $n+1$ 个方程，共有 $n$ 个未知数，其中一个方程有误差，求$n$个未知数中最大值的是第几个，方程的解是合法的当

1. 所有解都是正整数
2. 最大值有且仅有一个
3. 方程组其仅有一种解

若解不存在输出”illegal“

题解：

因为$1\le n\le 100$，所以可以枚举误差项，用剩下的 $n$ 个方程高斯消元判断解是否合法

#include<iostream>
#include<sstream>
#include<string>
#include<queue>
#include<map>
#include<unordered_map>
#include<set>
#include<vector>
#include<stack>
#include <utility>
#include<algorithm>
#include<cstdio>
#include<list>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<iomanip>
#include<time.h>
#include<random>
using namespace std;
#include<ext/pb_ds/priority_queue.hpp>
#include<ext/pb_ds/tree_policy.hpp>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/hash_policy.hpp>
using namespace __gnu_pbds;
#include<ext/rope>
using namespace __gnu_cxx;

#define int long long
#define PI acos(-1.0)
#define eps 1e-9
#define lowbit(a) ((a)&-(a))

const int mod = 1e9+7;
int qpow(int a,int b){
	int ans=1;
	while(b){
		if(b&1)ans=(ans*a)%mod;
		a=(a*a)%mod;
		b>>=1;
	}
	return ans;
}

const int INF = 0x3f3f3f3f;
const int N = 1e6+10;
double a[105][105],b[105][105];
int n;
int gauss(){
	for(int i=1;i<=n;i++){
		int p=i;
		for(int j=i+1;j<=n;j++)if(fabs(b[j][i])>fabs(b[p][i]))p=j;
		swap(b[i],b[p]);
		if(!b[i][i])return 0;
		for(int j=1;j<=n;j++){
			if(i==j)continue;
			double tmp=b[j][i]/b[i][i];
			for(int k=i+1;k<=n+1;k++)b[j][k]-=b[i][k]*tmp;
		}
	}
	int m=0,cnt=0;
	for(int i=1;i<=n;i++){
		b[i][n+1]/=b[i][i];
		if(b[i][n+1]<=0||(int)b[i][n+1]!=b[i][n+1])return 0;
		//判断所有未知数的解是否都是正整数	
	}
	for(int i=1;i<=n;i++)
		if(b[i][n+1]>=b[m][n+1]){
			if(b[i][n+1]>b[m][n+1])m=i,cnt=1;
			else cnt++;
			//找到所有未知数解的最大项，并判断最大项是否只有一个
		}
	if(cnt>1)return 0;
	return m;
}
#define endl '\n'
signed main(){
	std::ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	cin>>n;
	for(int i=1;i<=n+1;i++){
		int m; cin>>m;
		for(int j=1;j<=m;j++){
			int x; cin>>x;
			a[i][x]=1;
		}
		cin>>a[i][n+1];
	}
	int ans=0,flag=0;
	for(int i=1;i<=n+1;i++){//枚举误差行
		swap(a[i],a[n+1]);//把误差项放在n+1行，然后对前n行高斯消元
		for(int j=1;j<=n;j++)for(int k=1;k<=n+1;k++)b[j][k]=a[j][k]; 
		swap(a[i],a[n+1]);
		int res=gauss();
		if(res){
			if(flag){
				cout<<"illegal"<<endl; return 0;
			}
			ans=res; flag=1;//判断这个方程组是否只有一个解
		}
	}
	if(!flag)cout<<"illegal"<<endl;
	else cout<<ans<<endl;
}