Codeforces Round #545 (Div. 2) C. Skyscrapers 离散化+贪心
题目链接
给你一个n∗m的矩阵res,让你输出一个n∗m的矩阵a,这个矩阵满足:
ai,j代表第i行第j列的元素,通过减小res之后,使得第i行,第j列的所有元素的最大值
最小,并且相对大小不变。这个ai,j就是那个最大值
解法:对每一行每一列分别进行离散化,记录每个位置元素的相对大小,和最大元素的相对大小。
然后遍历每个元素进行讨论。分2种情况:
1.若当前位置在这一行这一列的相对大小一样,则,这个位置的答案为这一行,这一列的最大元素的相对大小中较大的那个。
2…若当前位置在这一行这一列的相对大小不一样,则要把那个较小的那个的最大元素的相对大小进行相应的加上几位。取一下行列的较大值即可。
具体细节见代码。
#include<bits/stdc++.h>
#define LL long long
#define fi first
#define se second
#define mp make_pair
#define pb push_back
using namespace std;
LL gcd(LL a,LL b){return b?gcd(b,a%b):a;}
LL lcm(LL a,LL b){return a/gcd(a,b)*b;}
LL powmod(LL a,LL b,LL MOD){LL ans=1;while(b){if(b%2)ans=ans*a%MOD;a=a*a%MOD;b/=2;}return ans;}
const int N = 1004;
int n,m,a[N][N],ans[N][N];
int row[N][N],cow[N][N],res[N],mar[N],mac[N];
int main(){
ios::sync_with_stdio(false);
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++){
for(int j=1;j<=m;j++){
res[j]=a[i][j];
}
sort(res+1,res+1+m);
int cnt=unique(res+1,res+1+m)-res-1;
for(int j=1;j<=m;j++){
int pos=lower_bound(res+1,res+1+cnt,a[i][j])-res;
row[i][j]=pos;
}
mar[i]=cnt;
}
for(int i=1;i<=m;i++){
for(int j=1;j<=n;j++){
res[j]=a[j][i];
}
sort(res+1,res+1+n);
int cnt=unique(res+1,res+1+n)-res-1;
for(int j=1;j<=n;j++){
int pos=lower_bound(res+1,res+1+cnt,a[j][i])-res;
cow[i][j]=pos;
}
mac[i]=cnt;
}
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
if(cow[j][i]==row[i][j]){
ans[i][j]=max(mac[j],mar[i]);
}else if(cow[j][i]>row[i][j]){
ans[i][j]=max(mac[j],mar[i]+abs(cow[j][i]-row[i][j]));
}else{
ans[i][j]=max(mac[j]+abs(cow[j][i]-row[i][j]),mar[i]);
}
}
}
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++)cout<<ans[i][j]<<' ';
cout<<endl;
}
return 0;
}