AcWing第67场周赛-4610. 列表排序

#include <bits/stdc++.h>
using namespace std;
#define de(x) cout<<x<<" ";
#define Pu puts("");
#define sf(x) scanf("%d",&x);
typedef long long ll;
const int N=2e2+10,mod=100003;
int n,m,ans;
int a[N][N];
int s[N];//对于每一行，想要达到的状态
int cha(){
    int cnt;//判断每一行是否在交换不大于一次的情况下，完成任务
    //对于列方面的操作，我们已经逆向的改变了s[i]数组
    for(int i=1;i<=n;i++){//意味着只要每一行的能变成当前状态
    //就可以在每行交换不超过一次，列交换不超过一次的要求下完成题目要求
        cnt=0;
        for(int j=1;j<=m;j++){
            if(s[j]!=a[i][j]) cnt++;
            if(cnt>2) return 0;
        }
    }
    return 1;
}
int fun(){
    for(int i=1;i<=m;i++) s[i]=i;
    if(cha()) return 1;

    for(int i=1;i<m;i++){
        for(int j=i+1;j<=m;j++){
            swap(s[i],s[j]);
            if(cha()) return 1;
            swap(s[i],s[j]);
        }
    }
    return 0;
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            sf(a[i][j])
        }
    }
    if(fun()) printf("YES\n");
    else printf("NO\n");
    return 0;
}