E - Obtain a Permutation
题意
给定一个n * m的矩阵,将变成变成有序矩阵,例如
3 2 1
1 2 3
4 5 6
将其变为
1 2 3
4 5 6
7 8 9
允许进行下面两种操作
1.每列可以向上移动m次(第一行变成最后一行的数字),修改m次。
2.可以直接更改位置上的数字,修改1次。
问将给定矩阵变成有序矩阵的最小修改次数。
思路
对每一列的最小修改次数进行贪心即可,看是移动后进行修改,还是直接修改成目标数字,移动n次会得到x个目标数字,n - x > 0 表示移动后来修改会得到更优的解。
具体思路:
怎样统计移动m次后得到几个目标数字,构造arr数组 = 目标数组 - 给定数组,根据上面一个矩阵。
可以变成数组arr
-2 0 2
3 3 3
3 3 3
观察每一行,会发现y = arr[i] / m,向上移动y次过后就能变成目标位置,然后这个题是说的循环移动。可能最后一行的数在第一行,求出的y是负数。既然是循环移动,那么加上取余就解决问题了,最终y变成 y = (arr[i] / m + n) % n。
统计移动y次得到的目标位置有多个,然后判断 arr[y] > y的话,那么选择移动。
时间复杂度 O(n * m)
AC代码
#include<bits\stdc++.h>
using namespace std;
const int maxn = 2e5 + 10;
int arr[maxn], temp[maxn];
void solve(){
int n, m, x;
scanf("%d%d", &n, &m);
for (int i = 1; i <= n * m; ++i){
scanf("%d", &x);
arr[i] = i - x;
}
int ans = 0;
for (int i = 1; i <= m; ++i){
for (int j = i; j <= n * m; j += m){
//如果这个数据不是m的倍数,或者是小于(-n*m),就一定会选择修改
if (arr[j] % m || arr[j] <= -(n * m)) continue;
int y = (arr[j] / m + n) % n;
temp[y]++;
}
//一列最多移动n次
int res = 0;
//temp表示移动i次有得到多少个目标位置
for (int j = 0; j < n; ++j){
if(temp[j] > j){
res = max (res, temp[j] - j);
}
temp[j] = 0;
}
//需要改的位置数
ans += (n - res);
}
printf("%d\n",ans);
}
int main(){
solve();
return 0;
}
/**
7 4
1 60204 147610 128455
77665 191006 94346 14148
145279 56741 146667 90808
149476 14 66548 61472
22435 36909 52368 79274
193242 75919 23 24
84776 141638 98306 75212
**/