codeforces 1294e Obtain a Permutation(暴力)

题目大意:

有一个二维矩阵Anm,每次我们可以选择其中一个数替换,或者选择一列让它进行向上的循环位移。

现在问我们,为了让矩阵变成:

1 2 ... ... m

m+1 m+2 ... ... 2m

... ... ... .,.. ...

1+(n-1)*m ...    m+(n-1)*m

需要至少进行几次操作。

n*m<=2e5.

解题思路:

为了讨论方遍,本题的所有下标从0开始,同时所有元素-1.

扫描二维码关注公众号,回复: 10400291 查看本文章

首先,它的操作决定了每列的操作是独立的,所以我们想到每列j开始遍历。接着,我们可以逐行i开始询问,首先判定这一个元素是否属于这一列,属于这一列的话有两个条件,第一 aij%m等于j 第二 它的位置必须小于n.

假设这个元素属于这一列,并且它的位置为pos. 那么所需要的循环上移的次数为:(i-pos+n)%n。每一列我们都记录下 循环上移对应次数的可以恢复多少个元素。比如这一列我循环上移3次的话,第一第二第三行元素恢复了,那么cnt[3][j]=3.

这一列的答案为:min_{i} n-cnt[i]+i 其中i:[0,n-1]

#include <bits/stdc++.h>
using namespace std;
int n,m;

int main(){
    cin>>n>>m;
    vector<vector<int>> gra(n+10,vector<int> (m+10,0));
    vector<vector<int>> cnt;
    cnt=gra;
    for(int i=0;i<n;i++)
        for(int j=0;j<m;j++){
            cin>>gra[i][j];
            gra[i][j]--;
        }
        int ans=0;
    for(int j=0;j<m;j++){
        for(int i=0;i<n;i++){
            int no=gra[i][j];
            if(no%m==j){
                int pos=no/m;
                if(pos<n){
                    cnt[(i-pos+n)%n][j]++;
                }
            }
        }
        int tmpans=1e9;
        for(int i=0;i<n;i++){
            tmpans=min(tmpans,n-cnt[i][j]+i);
        }
        ans+=tmpans;
    }
    cout<<ans<<endl;
	return 0;
}
发布了171 篇原创文章 · 获赞 4 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/FrostMonarch/article/details/104759183
今日推荐