4.求解线性方程组【算法赛】 - 蓝桥云课 (lanqiao.cn)
思路:
刚开始看到这题的时候还以为要用矩阵来做,看完题之后发现根本不用,因为x[i]只能由0和1构成,又因为a[i]已知,所以题目总共只有两种情况
一是:当x[1]为0时,x[2]以及x[i]全部可以由x[i]=a[i-1]-x[i-1]-x[i-2]递推出来;
二是:当x[1]为1时,思路同上;
本题最大的坑点就是:题目所描述的x[i]又0和1构成,所以不少人可能会觉得x[i]为1101,101,10......这种二进制的数,若正确理解题意之后,这题还是蛮简单的;
最后题目要求按字典序小的答案输出,那我们就先讨论x[1]=0的情况,若求不出1满足题意的解,则再求解x[1]=1时的方程组解,判断是否满足题意就看x[i]是否只为0或1;
C++代码:
#include <iostream>
using namespace std;
const int N=200000+5;
int n;
int a[N];
int x[N];
int flag;
int main()
{
// 请在此输入您的代码
cin>>n;
for(int i=1;i<=n;i++)
cin>>a[i];
x[1]=0;
for(int i=2;i<=n;i++){
x[i]=a[i-1]-x[i-1]-x[i-2];
if(x[i]!=0&&x[i]!=1){
flag=1;
x[1]=1;
break;
}
}
if(flag){
for(int i=2;i<=n;i++){
x[i]=a[i-1]-x[i-1]-x[i-2];
}
}
for(int i=1;i<=n;i++)
cout<<x[i]<<' ';
return 0;
}