2 seconds
256 megabytes
standard input
standard output
You are given a table consisting of n rows and m columns.
Numbers in each row form a permutation of integers from 1 to m.
You are allowed to pick two elements in one row and swap them, but no more than once for each row. Also, no more than once you are allowed to pick two columns and swap them. Thus, you are allowed to perform from 0 to n + 1 actions in total. Operations can be performed in any order.
You have to check whether it's possible to obtain the identity permutation 1, 2, ..., m in each row. In other words, check if one can perform some of the operation following the given rules and make each row sorted in increasing order.
The first line of the input contains two integers n and m (1 ≤ n, m ≤ 20) — the number of rows and the number of columns in the given table.
Each of next n lines contains m integers — elements of the table. It's guaranteed that numbers in each line form a permutation of integers from 1 to m.
If there is a way to obtain the identity permutation in each row by following the given rules, print "YES" (without quotes) in the only line of the output. Otherwise, print "NO" (without quotes).
2 4
1 3 2 4
1 3 4 2
YES
4 4
1 2 3 4
2 3 4 1
3 4 1 2
4 1 2 3
NO
3 6
2 1 3 4 5 6
1 2 4 3 5 6
1 2 3 4 6 5
YES
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <climits>
#include <cstring>
#include <string>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <vector>
#include <list>
#define rep(i,m,n) for(i=m;i<=n;i++)
#define rsp(it,s) for(set<int>::iterator it=s.begin();it!=s.end();it++)
#define mod 1000000007
#define inf 0x3f3f3f3f
#define vi vector<int>
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define ll long long
#define pi acos(-1.0)
#define pii pair<int,int>
#define Lson L, mid, ls[rt]
#define Rson mid+1, R, rs[rt]
const int maxn=1e5+;
using namespace std;
ll gcd(ll p,ll q){return q==?p:gcd(q,p%q);}
ll qpow(ll p,ll q){ll f=;while(q){if(q&)f=f*p;p=p*p;q>>=;}return f;}
inline ll read()
{
ll x=;int f=;char ch=getchar();
while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
return x*f;
}
int n,m,k,t,c[][],vis[][],num[];
struct node
{
int cnt,id;
bool operator<(const node&p)const
{
return cnt>p.cnt;
}
}ca[];
bool flag;
bool solve(int a,int b)
{
int d[][];
int i,j;
rep(i,,n)rep(j,,m)d[i][j]=c[i][j];
rep(i,,n)swap(d[i][a],d[i][b]);
rep(i,,n)
{
int now=;
rep(j,,m)if(d[i][j]!=j)now++;
if(now>)return false;
}
return true;
}
int main()
{
int i,j;
scanf("%d%d",&n,&m);
rep(i,,n)ca[i].id=i;
rep(i,,n)rep(j,,m)
{
scanf("%d",&c[i][j]);
if(c[i][j]!=j)ca[i].cnt++;
}
sort(ca+,ca+n+);
if(ca[].cnt<=)puts("YES");
else if(ca[].cnt>)puts("NO");
else
{
j=;
int pos=ca[].id;
rep(i,,m)if(c[pos][i]!=i)num[j++]=i;
rep(i,,)rep(j,i+,)
{
if(solve(num[i],num[j]))return *puts("YES");
}
puts("NO");
}
//system("Pause");
return ;
}