• KM算法模板


    POJ2195 KM算法

    View Code
    /*==================================================*\
    | POJ2195 边上有权值求取值最大(最小)的完全匹配
    \*==================================================*/
    int N,M, cntm,cnth;
    int slack[MM],lx[MM],ly[MM], pre[MM], maze[MM][MM];
    bool sx[MM],sy[MM];
    struct Point{ int x,y; }pointm[MM],pointh[MM];
    
    void get_data() {
         cntm=cnth=1;
         for(i=1;i<=N;i++) {
             scanf("%s",ch);
             for(j=0;ch[j];j++) {
                 if(ch[j]=='m') pointm[cntm].x=i,pointm[cntm++].y=j+1;
                 if(ch[j]=='H') pointh[cnth].x=i,pointh[cnth++].y=j+1;
             }
         } 
         for(i=1;i<cntm;i++) {
             for(j=1;j<cnth;j++) {
                 temp=f_abs(pointm[i].x-pointh[j].x)+f_abs(pointm[i].y-pointh[j].y);
                 maze[i][j]=-temp;  //求最小的完全匹配加上的是负值,求最大就加上正值 
             }
         }
    }
    bool dfs(int u) {
        int i,j,k,temp;
        sx[u]=true;
        for(i=1;i<cnth;i++) {
            if(sy[i])  continue;
            temp=lx[u]+ly[i]-maze[u][i];
            if(!temp) {
                 sy[i]=true; if(pre[i]==-1||dfs(pre[i])) { pre[i]=u; return true; }
            }
            else if(slack[i]>temp) slack[i]=temp;
        }
        return false;
    }
    
    int bestmatch() { //返回权值最大(最小的完全匹配)
        int i,j,k;
        for(i=1;i<cntm;i++) {
            lx[i]=-inf; ly[i]=0;
            for(j=1;j<cnth;j++) {
                if(lx[i]<maze[i][j]) lx[i]=maze[i][j];
            }
        }
        memset(pre,-1,sizeof(pre));
        for(i=1;i<cntm;i++) {
            for(j=0;j<cntm;j++) slack[j]=inf;
              while(true) {
                    memset(sx,false,sizeof(sx));
                    memset(sy,false,sizeof(sy));
                    if(dfs(i))  break;
                    int dx=inf;
                    for(j=1;j<cntm;j++) {
                         if(!sy[j]&&dx>slack[j]) dx=slack[j];
                    }
                    for(j=1;j<cntm;j++) {
                          if(sx[j]) lx[j]-=dx;
                          if(sy[j]) ly[j]+=dx;
                          if(!sy[j]) slack[j]-=dx; 
                    } 
              }
        }
        int sum=0;
        for(i=1;i<cntm;i++) sum-=maze[pre[i]][i]; //求最小的完全匹配加上的是负值,求最大就加上正值 
        return sum;
    }
  • 相关阅读:
    web网站的测试需考虑的问题
    颜色搭配
    30岁前不必在乎的30件事情
    数据库时代的终结
    实现IT创业的十三种模式分析
    Linux rpm 命令参数使用详解[介绍和应用]
    PHP中的正则表达式及模式匹配
    linux下使用yum安装Apache+php+Mysql+phpMyAdmin
    jQuery.extend函数详细用法
    PHP CURL HTTP 研究笔记
  • 原文地址:https://www.cnblogs.com/zhang1107/p/2814881.html
Copyright © 2020-2023  润新知