题目大意
给定一个
n×m
的01矩阵,每次能选择两个相邻的位置交换。第一行与最后一行,第一列与最后一列视为相邻。
先要将这个矩阵变成每行每列的1个数分别相等(同时满足或者不同时满足)。
求最小交换次数。
Data Constraint
n,m≤1000
题解
先统计矩阵中1的个数,根据是否是
n,m
的倍数来判断最终的局面是哪种情况。
然后考虑如何分配。首先,行和列是独立的,因为每次交换只会改变行或列中的一个的1的个数(即行和列的1的个数再一次交换中不会同时变化)。
所以现在的问题转化成,给定一个序列,每次可以将一个物品从一个位置转移到相邻的位置。使得最终每个位置物品数相等。求最小转移次数。
这是一个很经典的问题,记每个人的物品数-平均值的数为
ai
,我们要将
a
调整到全为0。我们假设第一个人给了第二个人
时间复杂度:
SRC
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std ;
#define N 1000 + 10
int Row[N] , Col[N] , a[N] , S[N] ;
int Mat[N][N] ;
int T , n , m ;
int Cnt , ans ;
void Solve( int *b , int st , int k ) {
for (int i = 1 ; i <= k ; i ++ ) {
a[i] = b[i] - st ;
S[i] = S[i-1] + a[i] ;
}
sort( S + 1 , S + k + 1 ) ;
int mid = S[(1+k)/2] ;
for (int i = 1 ; i <= k ; i ++ ) ans += abs( S[i] - mid ) ;
}
int main() {
scanf( "%d" , &T ) ;
for (int Case = 1 ; Case <= T ; Case ++ ) {
printf( "Case %d: " , Case ) ;
scanf( "%d%d" , &n , &m ) ;
memset( Row , 0 , sizeof(Row) ) ;
memset( Col , 0 , sizeof(Col) ) ;
Cnt = ans = 0 ;
for (int i = 1 ; i <= n ; i ++ ) {
scanf( "\n" ) ;
for (int j = 1 ; j <= m ; j ++ ) {
char ch ;
scanf( "%c" , &ch ) ;
if ( ch == '1' ) Mat[i][j] = 1 , Row[i] ++ , Col[j] ++ , Cnt ++ ;
else Mat[i][j] = 0 ;
}
}
if ( Cnt % n == 0 && Cnt % m == 0 ) printf( "both " ) ;
else if ( Cnt % n == 0 ) printf( "row " ) ;
else if ( Cnt % m == 0 ) printf( "column " ) ;
else { printf( "impossible\n" ) ; continue ; }
if ( Cnt % n == 0 ) Solve( Row , Cnt / n , n ) ;
if ( Cnt % m == 0 ) Solve( Col , Cnt / m , m ) ;
printf( "%d\n" , ans ) ;
}
return 0 ;
}
以上.