其他分享
首页 > 其他分享> > 一类老鼠进洞模型

一类老鼠进洞模型

作者:互联网

问题描述

给定一个序列,序列上有若干\(mouse\)和若干\(hole\),求一组最优的\(mouse\)和\(hole\)的匹配。
定义一只\(mouse\)跑到一个洞\(hole\)的代价为两点之间的距离。

问题一:洞有容量、有代价 ;

每一个洞不一定要进老鼠,但每一个老鼠一定要进一个洞。

结论一:匹配不会交叉。(显然)
结论二:对于一组匹配,老鼠和洞不会同时反悔。(从左往右进行匹配,同时反悔会产生交叉)

然后就可以用堆模拟费用流啦。
对老鼠和洞分别维护一个堆,设老鼠堆为\(Q_1\),洞的堆为\(Q_2\),表示增广集合。
方便起见先往\(Q_2\)中放一个代价为\(inf\)、容量为\(inf\)的洞。

代码实现相当简单:

Q2.push((Item){inf , 1000000000}) ; Ans = 0 ;
Item u ; ll t , w , fl , cnt ; 
for(int i = 1; i <= n; i ++) {
    if(!p[i].op) {
        u = Q2.top() ;
        Q2.pop() ;
        w = u.val + p[i].x ; Ans += w ; u.flow -- ;
        if(u.flow > 0) Q2.push(u) ;
        Q1.push((Item){- w - p[i].x , 1}) ; 
    }
    else {
        cnt = 0 ; 
        while(p[i].cap && !Q1.empty()) {
            u = Q1.top() ; 
            w = p[i].x + u.val + p[i].cost ; if(w >= 0) break ;
            Q1.pop() ;
            fl = min(p[i].cap , u.flow) ;
            Ans = Ans + 1ll * w * fl ;
            u.flow -= fl ;
            p[i].cap -= fl ; 
            cnt += fl ;
            if(u.flow > 0) Q1.push(u) ;
            Q2.push((Item){- w + p[i].cost - p[i].x , fl}) ;
        }
        if(p[i].cap > 0) Q2.push((Item){- p[i].x + p[i].cost , p[i].cap}) ;
        if(cnt > 0) Q1.push((Item){- p[i].cost - p[i].x , cnt}) ;
    }
}
cout << Ans << endl ; return 0 ;

问题二:洞有容量、有代价 ; 老鼠有容量、有代价

堆模拟费用流,但复杂度没有保证。
对老鼠和洞都进行两种反悔(即老鼠和洞都使用问题二中洞的反悔方式)。
常见转换有:若一个洞必须进老鼠、或者一只老鼠必须进洞,那么就把它的代价赋为\(-inf\)。
部分问题的特殊性质可以保证复杂度,其他情况下作为一个暴力算法还是很优秀的。

标签:老鼠,匹配,代价,模型,进洞,反悔,cost,fl
来源: https://www.cnblogs.com/GuessYCB/p/10578089.html