Codeforces 724B Batch Sort(暴力枚举)

题目链接：http://codeforces.com/problemset/problem/724/B

题目大意：给你一个n*m的矩阵，每次可以交换每行的任意两个数或者交换任意两列（只有一次交换机会）；最多有N+1次操作，如果操作后的矩阵每一行都是1~m的单调递增顺序，则输出YES，否则输出NO。

思路：暴力枚举法，先判断是否需要交换列，如果不需要直接输出YES，如果需要枚举交换任意一列，然后再判断。

AC代码：

#include<stdio.h>
#include<algorithm>
#include<iostream>
#include<string.h>
using namespace std;
int a[25][25];
int n,m;
bool judge()
{
    for(int i=1;i<=n;i++)
    {
        int co=0;
        for(int j=1;j<=m;j++)
        {
            if(a[i][j]!=j)
                co++;
        }
        if(co>2)//如果每行要交换的个数大于2，则需要交换列
            return false;
    }
    return true;//否则，不需要。
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        scanf("%d",&a[i][j]);
    if(judge())
    {
        printf("YES\n");
        return 0;
    }
    for(int i=1;i<m;i++)
    {
        for(int j=i+1;j<=m;j++)
        {
            //两层for循环枚举要交换的列
            int flag=1;
            for(int k=1;k<=n;k++)
                swap(a[k][i],a[k][j]);//交换i，j两列
            if(judge())
            {
                printf("YES\n");
                return 0;
            }
            for(int k=1;k<=n;k++)
                swap(a[k][i],a[k][j]);
        }
    }
    printf("NO\n");
    return 0;
}

原创作品，如能转载，不胜荣幸！