[ARC115B] Plus Matrix 的题解
题目大意
给你一个 的数组 ,,求 数组与 数组,不保证有解,其中 ,而且 都是非负整数。
思路
因为 。
所以 。
将我们一行依次相减,就得到了 的一组关系,其实就是 个等式。
显然,对于所有的 这 个等式应该都是满足的。如果不能满足这个条件,那么就是无解的。
因为 都是非负整数,所以我们需要找到一组解使上面的等式全部满足。为了让答案在最后好处理,我们可以先找到一组最小解。
因为上面的方程对于所有的 都满足,所以我们只需要考虑一行就可以了,为了便于讨论,我选取了第 行。得到第 行最小的元素 ,所有 ,对于 数组也是如此。
因为我们找到的是最小解,所有有可能会出现 的情况。因为我们已经得到了 数组之间的关系,那么我们就可以将 数组全部加 。
AC Code
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=505;
int n,c[N][N],a[N],b[N];
signed main(){
ios::sync_with_stdio(false);
cin.tie(nullptr);
cin>>n;
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
cin>>c[i][j];
}
}
for(int i=1;i<n;i++){
for(int j=1;j<n;j++){
if(c[i][j]-c[i][j+1]!=c[i+1][j]-c[i+1][j+1]){
cout<<"No";
return 0;
}
}
}
for(int j=1;j<n;j++){
for(int i=1;i<n;i++){
if(c[i][j]-c[i][j+1]!=c[i+1][j]-c[i+1][j+1]){
cout<<"No";
return 0;
}
}
}
int mmin=INT_MAX;
for(int i=1;i<=n;i++){
mmin=min(c[1][i],mmin);
}
for(int i=1;i<=n;i++){
b[i]=c[1][i]-mmin;
}
mmin=INT_MAX;
for(int i=1;i<=n;i++){
mmin=min(c[i][1],mmin);
}
for(int i=1;i<=n;i++){
a[i]=c[i][1]-mmin;
}
cout<<"Yes\n";
int add=c[1][1]-a[1]-b[1];
for(int i=1;i<=n;i++){
cout<<a[i]<<" ";
}
cout<<'\n';
for(int i=1;i<=n;i++){
cout<<b[i]+add<<" ";
}
return 0;
}