Posted on 2011-07-23 22:14
Mato_No1 閱讀(7635)
評(píng)論(9) 編輯 收藏 引用 所屬分類:
圖算法
【背景（神犇不要鄙視）】
本沙茶今年AHOI的時(shí)候，遇到裸的最佳匹配的題，竟然把KM算法搞忘了，幸虧是WJMZBMR神犇保佑，臨時(shí)亂弄一通，想起來(lái)了……這MS反映出了本沙茶以前在看某些經(jīng)典算法的時(shí)候看得不深，木有理解透徹……
前幾天又遇到一道最佳匹配的題，發(fā)現(xiàn)KM算法竟然又忘了……米辦法，只有把這個(gè)搞死人的算法的具體過(guò)程重新看了一遍，終于懂了……
【KM算法及其具體過(guò)程】
（1）可行點(diǎn)標(biāo)：每個(gè)點(diǎn)有一個(gè)標(biāo)號(hào)，記lx[i]為X方點(diǎn)i的標(biāo)號(hào)，ly[j]為Y方點(diǎn)j的標(biāo)號(hào)。如果對(duì)于圖中的任意邊(i, j, W)都有l(wèi)x[i]+ly[j]>=W，則這一組點(diǎn)標(biāo)是可行的。特別地，對(duì)于lx[i]+ly[j]=W的邊(i, j, W)，稱為可行邊；
（2）KM算法的核心思想就是通過(guò)修改某些點(diǎn)的標(biāo)號(hào)（但要滿足點(diǎn)標(biāo)始終是可行的），不斷增加圖中的可行邊總數(shù)，直到圖中存在僅由可行邊組成的完全匹配為止，此時(shí)這個(gè)匹配一定是最佳的（因?yàn)橛煽尚悬c(diǎn)標(biāo)的的定義，圖中的任意一個(gè)完全匹配，其邊權(quán)總和均不大于所有點(diǎn)的標(biāo)號(hào)之和，而僅由可行邊組成的完全匹配的邊權(quán)總和等于所有點(diǎn)的標(biāo)號(hào)之和，故這個(gè)匹配是最佳的）。一開始，求出每個(gè)點(diǎn)的初始標(biāo)號(hào)：lx[i]=max{e.W|e.x=i}（即每個(gè)X方點(diǎn)的初始標(biāo)號(hào)為與這個(gè)X方點(diǎn)相關(guān)聯(lián)的權(quán)值最大的邊的權(quán)值），ly[j]=0（即每個(gè)Y方點(diǎn)的初始標(biāo)號(hào)為0）。這個(gè)初始點(diǎn)標(biāo)顯然是可行的，并且，與任意一個(gè)X方點(diǎn)關(guān)聯(lián)的邊中至少有一條可行邊；
（3）然后，從每個(gè)X方點(diǎn)開始DFS增廣。DFS增廣的過(guò)程與最大匹配的Hungary算法基本相同，只是要注意兩點(diǎn)：一是只找可行邊，二是要把搜索過(guò)程中遍歷到的X方點(diǎn)全部記下來(lái)（可以用vst搞一下），以進(jìn)行后面的修改；
（4）增廣的結(jié)果有兩種：若成功（找到了增廣軌），則該點(diǎn)增廣完成，進(jìn)入下一個(gè)點(diǎn)的增廣。若失敗（沒(méi)有找到增廣軌），則需要改變一些點(diǎn)的標(biāo)號(hào)，使得圖中可行邊的數(shù)量增加。方法為：將所有在增廣軌中（就是在增廣過(guò)程中遍歷到）的X方點(diǎn)的標(biāo)號(hào)全部減去一個(gè)常數(shù)d，所有在增廣軌中的Y方點(diǎn)的標(biāo)號(hào)全部加上一個(gè)常數(shù)d，則對(duì)于圖中的任意一條邊(i, j, W)（i為X方點(diǎn)，j為Y方點(diǎn)）：
<1>i和j都在增廣軌中：此時(shí)邊(i, j)的(lx[i]+ly[j])值不變，也就是這條邊的可行性不變（原來(lái)是可行邊則現(xiàn)在仍是，原來(lái)不是則現(xiàn)在仍不是）；
<2>i在增廣軌中而j不在：此時(shí)邊(i, j)的(lx[i]+ly[j])的值減少了d，也就是原來(lái)這條邊不是可行邊（否則j就會(huì)被遍歷到了），而現(xiàn)在可能是；
<3>j在增廣軌中而i不在：此時(shí)邊(i, j)的(lx[i]+ly[j])的值增加了d，也就是原來(lái)這條邊不是可行邊（若這條邊是可行邊，則在遍歷到j(luò)時(shí)會(huì)緊接著執(zhí)行DFS(i)，此時(shí)i就會(huì)被遍歷到），現(xiàn)在仍不是；
<4>i和j都不在增廣軌中：此時(shí)邊(i, j)的(lx[i]+ly[j])值不變，也就是這條邊的可行性不變。
這樣，在進(jìn)行了這一步修改操作后，圖中原來(lái)的可行邊仍可行，而原來(lái)不可行的邊現(xiàn)在則可能變?yōu)榭尚羞叀Ｄ敲磀的值應(yīng)取多少？顯然，整個(gè)點(diǎn)標(biāo)不能失去可行性，也就是對(duì)于上述的第<2>類邊，其lx[i]+ly[j]>=W這一性質(zhì)不能被改變，故取所有第<2>類邊的(lx[i]+ly[j]-W)的最小值作為d值即可。這樣一方面可以保證點(diǎn)標(biāo)的可行性，另一方面，經(jīng)過(guò)這一步后，圖中至少會(huì)增加一條可行邊。
（5）修改后，繼續(xù)對(duì)這個(gè)X方點(diǎn)DFS增廣，若還失敗則繼續(xù)修改，直到成功為止；
下面分析整個(gè)算法的時(shí)間復(fù)雜度：每次修改后，圖中至少會(huì)增加一條可行邊，故最多增廣M次、修改M次就可以找到僅由可行邊組成的完全匹配（除非圖中不存在完全匹配，這個(gè)可以通過(guò)預(yù)處理得到），故整個(gè)算法的時(shí)間復(fù)雜度為O(M * (N + 一次修改點(diǎn)標(biāo)的時(shí)間))。而一次修改點(diǎn)標(biāo)的時(shí)間取決于計(jì)算d值的時(shí)間，如果暴力枚舉計(jì)算，這一步的時(shí)間為O(M)，優(yōu)化：可以對(duì)每個(gè)Y方點(diǎn)設(shè)立一個(gè)slk值，表示在DFS增廣過(guò)程中，所有搜到的與該Y方點(diǎn)關(guān)聯(lián)的邊的(lx+ly-W)的最小值（這樣的邊的X方點(diǎn)必然在增廣軌中）。每次DFS增廣前，將所有Y方點(diǎn)的slk值設(shè)為+∞，若增廣失敗，則取所有不在增廣軌中的Y方點(diǎn)的slk值的最小值為d值。這樣一次修改點(diǎn)標(biāo)的時(shí)間降為O(N)，總時(shí)間復(fù)雜度降為O(NM)。
需要注意的一點(diǎn)是，在增廣過(guò)程中需要記下每個(gè)X、Y方點(diǎn)是否被遍歷到，即fx[i]、fy[j]。因此，在每次增廣前（不是對(duì)每個(gè)X方點(diǎn)增廣前）就要將所有fx和fy值清空。
代碼：
void init_d()
{
re(i, n) E[i].pre = E[i].next = i; m = n;
}
void add_edge(int a, int b, int len)
{
E[m].a = a; E[m].b = b; E[m].len = len; E[m].pre = E[a].pre; E[m].next = a; E[a].pre = m; E[E[m].pre].next = m++;
}
inline int dist(int x, int y, int x0, int y0)
{
return abs(x - x0) + abs(y - y0);
}
bool dfs(int x)
{
int y, x0; fx[x] = _FLAG;
for (int p=E[x].next; p != x; p=E[p].next) {
y = E[p].b;
if (lx[x] + ly[y] > E[p].len) {
if (lx[x] + ly[y] - E[p].len < slk[y]) slk[y] = lx[x] + ly[y] - E[p].len;
} else if (fy[y] != _FLAG) {
fy[y] = _FLAG; x0 = z[y];
if (x0 == -1 || dfs(x0)) {z[y] = x; return 1;}
}
}
return 0;
}
void solve()
{
re(i, n) {
lx[i] = -INF;
for (int p=E[i].next; p != i; p=E[p].next) if (E[p].len > lx[i]) lx[i] = E[p].len;
}
re(i, n0) {ly[i] = 0; z[i] = -1;}
int d;
re(i, n) {
re(j, n0) slk[i] = INF; _FLAG++;
while (!dfs(i)) {
d = INF; re(j, n0) if (fy[j] != _FLAG && slk[j] < d) d = slk[j];
re(j, n) if (fx[j] == _FLAG) lx[j] -= d;
re(j, n0) {slk[j] = INF; if (fy[j] == _FLAG) ly[j] += d;}
_FLAG++;
}
}
res = 0; re(i, n) res += lx[i]; re(i, n0) res += ly[i];
}