题目
https://codeforces.com/contest/1294/problem/E
题意
一个矩阵,有一个原始的次序。
现在给你一个打乱的矩阵,要你用最少的操作次数复原。
你有两种操作:
1、把某个数字直接变成任意一个数字
2、把某一列整体向上移动一格(循环移动)
思路
首先我们知道,每一列之间不互相影响,也就是说,我们可以一列一列解决问题,得到答案。
对于当前列,我们希望得到把这一列复原所需要的最少操作次数。
我们可以想到,如果把当前列向上移动 i 次,此时有 a[i] 个数字在正确位置上,那么就有 n - a[i] 个数字在不正确的位置上,也就是说,我对于这一列把它复原所需要进行的操作次数是 i + n - a[i] ,其中n - a[i]的修改方式是直接修改成正确的数字。
我们取最小值即可得到该列的答案,然后拓展到所有列,求和即为答案。
关键在于我要如何用a[i]来记录移动了 i 次时,该列上位于正确位置的数字的个数。
一开始我的想法是,直接移动一次,然后遍历看有多少个在正确的位置上。
但是这样总的时间复杂度为O(n * n * m)会超时
那么可以进行更优的统计方法,也就是,对于每一个数字,直接计算它向上移动多少次后才会位于正确的位置上。在该位置上+1,这样就能在O(n)的时间复杂度内得到a[i]数组,总的时间复杂度为O(n * m)
代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int INF = 0x3f3f3f3f;
const int maxn = 2e5 + 10;
vector <int> vec[maxn];
int a[maxn];
int main()
{
int n, m;
scanf("%d %d", &n, &m);
for(int i = 1; i <= n; i++)
{
vec[i].push_back(0);
for(int j = 1; j <= m; j++)
{
int x;
scanf("%d", &x);
vec[i].push_back(x);
}
}
int ans = 0;
for(int j = 1; j <= m; j++)
{
int mini = maxn;
for(int i = 0; i < n; i++){
a[i] = 0;
}
for(int i = 1; i <= n; i++)
{
if(vec[i][j] <= n * m && (vec[i][j] - j) % m == 0)//在范围内并且是这一列的
{
int c = (vec[i][j] - j) / m + 1;//这个数字应该在的行
a[(i - c + n) % n]++;//a[i]为移动i次后在正确位置上的数字的个数
}
}
for(int i = 0; i < n; i++)
{
mini = min(mini, i + n - a[i]);//i为移动的次数, n - a[i]为移动后不在正确位置上的数字
}
ans += mini;
}
printf("%d\n", ans);
return 0;
}