https://vjudge.net/problem/CodeForces-1294E
题目大意:给一个 n ∗ m n*m n∗m的矩阵,每次操作可以选取任意一个位置的数使其变成 [ 1 , n ∗ m ] [1,n*m] [1,n∗m]的任意一个值,或者选取某一列,进行上移操作(具体看图吧,不太好解释)。问最少经过几次操作,可以得到满足题意的矩阵。
思路:其实就是个贪心,但是实现上可能有点麻烦。因为列与列是独立的,所以可以一列一列的考虑。我们用 c n t [ i ] cnt[i] cnt[i]记录在当前列的情况下,整体上移 i i i个单位后满足题意的最大元素个数。假设当前在第 i i i列,枚举该列的所有元素,如果 m a t r i x [ j ] [ i ] % m = i − 1 matrix[j][i]\%m=i-1 matrix[j][i]%m=i−1的话,再令 t m p = m a t r i x [ j ] [ i ] / m tmp=matrix[j][i]/m tmp=matrix[j][i]/m,如果 t m p < n tmp<n tmp<n的话,说明这个数可以通过上移操作到恰当的位置,且需要上移 ( j − t m p + n ) % n (j-tmp+n)\%n (j−tmp+n)%n个单位,那么我们令 c n t [ ( j − t m p + n ) % n ] cnt[(j-tmp+n)\%n] cnt[(j−tmp+n)%n]自增。枚举完之后,初始化 a n s = n ans=n ans=n,然后遍历 c n t [ 0 … n − 1 ] cnt[0…n-1] cnt[0…n−1],取 a n s = m i n ( a n s , n − c n t [ i ] + i ) ans=min(ans,n-cnt[i]+i) ans=min(ans,n−cnt[i]+i)。解释一下后半部分怎么来的, n − c n t [ i ] n-cnt[i] n−cnt[i]是需要变化值的贡献, i i i是需要上移的贡献,因此消耗的步数为 n − c n t [ i ] + i n-cnt[i]+i n−cnt[i]+i。
#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
int n,m;
vector<int> vec[maxn];
int main()
{
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++)
{
vec[i].resize(m);
for(int j=0;j<m;j++)
{
scanf("%d",&vec[i][j]);
--vec[i][j];
}
}
int ans=0;
for(int i=0;i<m;i++)
{
int res=n;
vector<int> cnt(n+5);
for(int j=0;j<n;j++)
{
if(vec[j][i]%m==i)
{
int tmp=vec[j][i]/m;
if(tmp<n)
++cnt[(j-tmp+n)%n];
}
}
for(int j=0;j<n;j++)
res=min(res,n-cnt[j]+j);
ans+=res;
}
printf("%d\n",ans);
return 0;
}