hdu 2819 Swap

題目鏈接:http://acm.hdu.edu.cn/showproblem.php?pid=2819

題目大意:通過行列交換,讓矩陣中主對角線的值全爲1。

題目分析:只交換行或者只交換列都是可以換出來的,此題爲二分圖的匹配問題,x集合爲行號,y集合爲列號,用匈牙利算法完成匹配。

代碼參考:

#include<map>
#include<cstdio>
#include<cstring>
using namespace std;
const int N = 200;
bool vis[N];//記錄y中節點是否使用
int cx[N], cy[N], g[N][N], ansx[N], ansy[N];
int n;//二分圖中x和y中點的數目
bool dfs(int u)// 匈牙利算法
{
	for (int v = 1; v <= n; ++v)
	{
		if (g[u][v] && !vis[v])//如果節點v與u相鄰並且未被查找過
		{
			vis[v] = true;//標記v爲已查找過

			if (cy[v] == -1 || dfs(cy[v]))//如果v未在前一個匹配cy中,或者v在匹配cy中,但是從與v相鄰的節點出發可以有增廣路徑  
			{//記錄查找成功記錄,更新匹配cx,cy 
				cx[u] = v;
				cy[v] = u;
				return true;//返回查找成功 
			}
		}
	}

	return false;// 不存在增廣路,返回失敗
}
int MaxMatch()
{
	int res = 0;//找到多少匹配
	//設置初始匹配
	memset(cx, -1, sizeof(cx));
	memset(cy, -1, sizeof(cy));

	for (int i = 1; i <= n; ++i)
	{
		memset(vis, 0, sizeof(vis)); //清空上次搜索時的標記  

		if (dfs(i)) //從節點i嘗試擴展 
		{
			res++;//找到新的匹配
		}
		else
		{
			break;
		}
	}

	return res;
}
int main()
{
	int m, t, i, j, k;

	while (~scanf("%d", &n))
	{ //構造二部圖
		for (i = 1; i <= n; ++i)
		{
			for (j = 1; j <= n; ++j)
			{
				scanf("%d", &g[i][j]);//記錄連接x和y的邊,如果i和j之間有邊則爲1,否則爲0
			}
		}

		if (MaxMatch() != n)
		{
			puts("-1");
			continue;
		}

		int len = 0;

		for (i = 1; i <= n; ++i)//選擇排序
		{
			int p = i;

			for (j = i; j <= n; ++j)
			{
				if (cy[p] > cy[j])//列對應的行從小到大排序
				{
					p = j;
				}
			}

			if (p == i)
			{
				continue;
			}
			//交換的列,就是要輸出的
			ansx[++len] = i;
			ansy[len] = p;
			swap(cy[p], cy[i]);
		}

		printf("%d\n", len);

		for (i = 1; i <= len; ++i)
		{
			printf("C %d %d\n", ansx[i], ansy[i]);
		}
	}

	return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章