cogs 290. [CTSC2000] 丘比特的烦恼

290. [CTSC2000] 丘比特的烦恼

★★★   输入文件:cupid.in   输出文件:cupid.out   简单对比
时间限制:1 s   内存限制:128 MB

随着社会的不断发展,人与人之间的感情越来越功利化。最近,爱神丘比特发现,爱情也已不再是完全纯洁的了。这使得丘比特很是苦恼,他越来越难找到合适的男女,并向他们射去丘比特之箭。于是丘比特千里迢迢远赴中国,找到了掌管东方人爱情的神——月下老人,向他求教。
月下老人告诉丘比特,纯洁的爱情并不是不存在,而是他没有找到。在东方,人们讲究的是缘分。月下老人只要做一男一女两个泥人,在他们之间连上一条红线,那么它们所代表的人就会相爱——无论他们身处何地。而丘比特的爱情之箭只能射中两个距离相当近的人,选择的范围自然就小了很多,不能找到真正的有缘人。
丘比特听了月下老人的解释,茅塞顿开,回去之后用了人间的最新科技改造了自己的弓箭,使得丘比特之箭的射程大大增加。这样,射中有缘人的机会也增加了不少。
情人节(Valentine's day)的午夜零时,丘比特开始了自己的工作。他选择了一组数目相等的男女,感应到他们互相之间的缘分大小,并依此射出了神箭,使他们产生爱意。他希望能选择最好的方法,使被他选择的每一个人被射中一次,且每一对被射中的人之间的缘分的和最大。
当然,无论丘比特怎么改造自己的弓箭,总还是存在缺陷的。首先,弓箭的射程尽管增大了,但毕竟还是有限的,不能像月下老人那样,做到“千里姻缘一线牵 ”。其次,无论怎么改造,箭的轨迹终归只能是一条直线,也就是说,如果两个人之间的连线段上有别人,那么莫不可向他们射出丘比特之箭,否则,按月下老人的话,就是“乱点鸳鸯谱”了。
作为一个凡人,你的任务是运用先进的计算机为丘比特找到最佳的方案。

输入文件格式:
输入文件第一行为正整数k,表示丘比特之箭的射程,第二行为正整数n(n<30),随后有2n行,表示丘比特选中的人的信息,其中前n行为男子,后n行为女子。每个人的信息由两部分组成:他的姓名和他的位置。姓名是长度小于20且仅包含字母的字符串,忽略大小写的区别,位置是由一对整数表示的坐标,它们之间用空格分隔。格式为x y Name。输入文件剩下的部分描述了这些人的缘分。每一行的格式为Name1 Name2 p。Name1和Name2为有缘人的姓名,p是他们之间的缘分值(p为小于等于255的正整数)。以一个End作为文件结束标志。每两个人之间的缘分如果被描述多次,以最后一次为准。如果没有被描述,则说明他们缘分值为1。

输出文件格式:
输出文件仅一个正整数,表示每一对被射中的人之间的缘分的总和。这个和应当是最大的。

输入样例
2
3
0 0 Adam
1 1 Jack
0 2 George
1 0 Victoria
0 1 Susan
1 2 Cathy
Adam Cathy 100
Susan George 20
George Cathy 40
Jack Susan 5
Cathy Jack 30
Victoria Jack 20
Adam Victoria 15
End

输出样例
65

思路:先根据题目要求建图,然后跑KM算法。

错因:未知。不知道为什么就wa了。自测数据全部正确没交上评测结果不相同。

KM算法的代码。

#include<iostream>
#include<map>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct nond{
    int x,y;
}pe[70];
map<string,int>ma;
int k,n,ans,love[70][70],vis_boy[70],slack[70];
int match[70],ex_boy[70],ex_girl[70],vis_girl[70];
int judge(char s[],char ss[],int x){
    int s1=ma[s],s2=ma[ss];
    int x1=pe[s1].x-pe[s2].x;
    int y1=pe[s1].y-pe[s2].y;
    if(x1*x1+y1*y1>k*k)    return true;
    else
        for(int i=1;i<=2*n;i++){
            if(i==s1||i==s2)    continue;
            int x2=pe[s1].x-pe[i].x;
            int y2=pe[s1].y-pe[i].y;
            if(x1*y2-x2*y1!=0)    continue;
            if(pe[i].x>pe[s1].x&&pe[i].x<pe[s2].x&&pe[i].y>pe[s1].y&&pe[i].y<pe[s2].y
             ||pe[i].x>pe[s2].x&&pe[i].x<pe[s1].x&&pe[i].y>pe[s2].y&&pe[i].y<pe[s1].y
             ||pe[i].x>pe[s1].x&&pe[i].x<pe[s2].x&&pe[i].y>pe[s2].y&&pe[i].y<pe[s1].y
             ||pe[i].x>pe[s2].x&&pe[i].x<pe[s1].x&&pe[i].y>pe[s1].y&&pe[i].y<pe[s2].y)
                 return true;
            else return false;
        }
}
bool dfs(int girl){
    vis_girl[girl]=true;
    for(int boy=0;boy<n;boy++){
        if(vis_boy[boy])    continue;
        int now=ex_girl[girl]+ex_boy[boy]-love[girl][boy];
        if(now==0){
            vis_boy[boy]=true;
            if(match[boy]==-1||dfs(match[boy])){
                match[boy]=girl;
                return true;
            }
        }
        else    slack[boy]=min(slack[boy],now);
    }
    return false;
}
int KM(){
    memset(match,-1,sizeof(match));
    memset(ex_boy,0,sizeof(ex_boy));
    for(int i=0;i<n;i++){
        ex_girl[i]=love[i][0];
        for(int j=1;j<n;j++)
            ex_girl[i]=max(ex_girl[i],love[i][j]);
    }
    for(int i=0;i<n;i++){
        memset(slack,0x3f3f3f3f,sizeof(slack));
        while(1){
            memset(vis_boy,0,sizeof(vis_boy));
            memset(vis_girl,0,sizeof(vis_girl));
            if(dfs(i))    break;
            int d=0x3f3f3f3f;
            for(int j=0;j<n;j++)
                if(!vis_boy[j])    d=min(d,slack[j]);
            for(int j=0;j<n;j++){
                if(vis_girl[j])    ex_girl[j]-=d;
                if(vis_boy[j])    ex_boy[j]+=d;
                else    slack[j]-=d;
            }
        }
    }
    int rest=0;
    for(int i=0;i<n;i++)
        rest+=love[match[i]][i];
    return rest;
}
int main(){
    freopen("cupid.in","r",stdin);
    freopen("cupid.out","w",stdout);
    scanf("%d%d",&k,&n);
    for(int i=1;i<=2*n;i++){
        int x,y;char s[110];
        cin>>x>>y;cin>>s;
        for(int j=0;j<strlen(s);j++)
            if(s[j]>='A'&&s[j]<='Z')    s[j]+=32;
        pe[i].x=x;pe[i].y=y;
        ma[s]=i;
    }
    char s[110],ss[110];int x;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            love[i][j]=1;
    while(cin>>s){
        if(s[0]=='E'&&s[1]=='n'&&s[2]=='d')
            break;
        cin>>ss>>x;
        for(int j=0;j<strlen(s);j++)
            if(s[j]>='A'&&s[j]<='Z')    s[j]+=32;
        for(int j=0;j<strlen(ss);j++)
            if(ss[j]>='A'&&ss[j]<='Z')    ss[j]+=32;
        if(!judge(s,ss,x)){
            int s1=ma[s];
            int s2=ma[ss];
            if(s2>n)    swap(s1,s2);
            love[s1%n][s2%n]=x;
        }
    }
    ans=KM();
    cout<<ans;
    return 0;
}

 

posted @ 2017-08-24 16:17  一蓑烟雨任生平  阅读(145)  评论(0编辑  收藏  举报