hdu 2813(KM+map)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2813

思路：用STL的map把字符串映射为序号。。。差点就超时了。。。orz。。。弱渣不敢写字典树啊！！！

#include<iostream>
#include<map>
#include<string>
const int MAXN=222;
const int inf=1<<30;
using namespace std;
int n,m,k;
int match[MAXN];
int lx[MAXN],ly[MAXN];
int Map[MAXN][MAXN];
bool visitx[MAXN],visity[MAXN];

int Hungary(int u){
    visitx[u]=true;
    for(int i=1;i<=m;i++){
        if(!visity[i]&&lx[u]+ly[i]==Map[u][i]){
            visity[i]=true;
            if(match[i]==-1||Hungary(match[i])){
                match[i]=u;
                return true;
            }
        }
    }
    return false;
}


void KM_prefect_match(){
    int tmp;
    for(int i=1;i<=n;i++){
        lx[i]=-inf;
    }
    memset(ly,0,sizeof(ly));
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            lx[i]=max(lx[i],Map[i][j]);
        }
    }
    for(int i=1;i<=n;i++)
    {
        while(1){
            memset(visitx,false,sizeof(visitx));
            memset(visity,false,sizeof(visity));
            if(Hungary(i))
                break;
            else {
                tmp=inf;
                for(int j=1;j<=n;j++)if(visitx[j]){
                    for(int k=1;k<=m;k++){
                        if(!visity[k]&&tmp>lx[j]+ly[k]-Map[j][k]){
                            tmp=lx[j]+ly[k]-Map[j][k];
                        }
                    }
                }
                for(int j=1;j<=n;j++){
                    if(visitx[j])
                        lx[j]-=tmp;
                }
                for(int j=1;j<=m;j++){
                    if(visity[j])
                        ly[j]+=tmp;
                }
            }
        }
    }
}

int main(){
    while(~scanf("%d%d%d",&n,&m,&k)){
        for(int i=1;i<=n;i++){
            for(int j=1;j<=m;j++){
                Map[i][j]=-inf;
            }
        }
        memset(match,-1,sizeof(match));
        map<string,int>mp1,mp2;
        char s1[22],s2[22];
        int w,p1=1,p2=1;
        for(int i=1;i<=k;i++){
            scanf("%s%s%d",s1,s2,&w);
            string str1(s1),str2(s2);
            if(mp1[str1]==0){
                mp1[str1]=p1++;
            }
            if(mp2[str2]==0){
                mp2[str2]=p2++;
            }
            Map[mp1[str1]][mp2[str2]]=-w;
        }
        KM_prefect_match();
        int ans=0;
        for(int i=1;i<=m;i++){
            if(match[i]!=-1&&Map[match[i]][i]!=-inf)///必不可少的
                ans+=Map[match[i]][i];
        }
        printf("%d\n",-ans);
    }
    return 0;
}