PTA L2-044 大众情人

写代码的七柒

于 2024-04-02 21:22:24 发布

阅读量998

点赞数 25

分类专栏： PTA天梯赛文章标签：算法 c++ floyd

本文链接：https://blog.csdn.net/m0_75081848/article/details/137290758

版权

PTA天梯赛专栏收录该内容

79 篇文章 13 订阅

订阅专栏

人与人之间总有一点距离感。我们假定两个人之间的亲密程度跟他们之间的距离感成反比，并且距离感是单向的。例如小蓝对小红患了单相思，从小蓝的眼中看去，他和小红之间的距离为 1，只差一层窗户纸；但在小红的眼里，她和小蓝之间的距离为 108000，差了十万八千里…… 另外，我们进一步假定，距离感在认识的人之间是可传递的。例如小绿觉得自己跟小蓝之间的距离为 2，则即使小绿并不直接认识小红，我们也默认小绿早晚会认识小红，并且因为跟小蓝很亲近的关系，小绿会觉得自己跟小红之间的距离为 1+2=3。当然这带来一个问题，如果小绿本来也认识小红，或者他通过其他人也能认识小红，但通过不同渠道推导出来的距离感不一样，该怎么算呢？我们在这里做个简单定义，就将小绿对小红的距离感定义为所有推导出来的距离感的最小值。

一个人的异性缘不是由最喜欢他/她的那个异性决定的，而是由对他/她最无感的那个异性决定的。我们记一个人 i 在一个异性 j 眼中的距离感为 Dij；将 i 的“异性缘”定义为 1/maxj∈S(i){Dij}，其中 S(i) 是相对于 i 的所有异性的集合。那么“大众情人”就是异性缘最好（值最大）的那个人。

本题就请你从给定的一批人与人之间的距离感中分别找出两个性别中的“大众情人”。

输入格式：

输入在第一行中给出一个正整数 N（≤500），为总人数。于是我们默认所有人从 1 到 N 编号。

随后 N 行，第 i 行描述了编号为 i 的人与其他人的关系，格式为：

性别 K 朋友1:距离1 朋友2:距离2 …… 朋友K:距离K

其中 性别 是这个人的性别，F 表示女性，M 表示男性；K（<N 的非负整数）为这个人直接认识的朋友数；随后给出的是这 K 个朋友的编号、以及这个人对该朋友的距离感。距离感是不超过 106 的正整数。

题目保证给出的关系中一定两种性别的人都有，不会出现重复给出的关系，并且每个人的朋友中都不包含自己。

输出格式：

第一行给出自身为女性的“大众情人”的编号，第二行给出自身为男性的“大众情人”的编号。如果存在并列，则按编号递增的顺序输出所有。数字间以一个空格分隔，行首尾不得有多余空格。

输入样例：

6
F 1 4:1
F 2 1:3 4:10
F 2 4:2 2:2
M 2 5:1 3:2
M 2 2:2 6:2
M 2 3:1 2:5

输出样例：

2 3
4

测试点：

测试点4，检查floyd算法是否写错，注意floyd算法循环的顺序。

以下测试点信息来源于这篇博客L2-044 大众情人测试点(1和3)-CSDN博客

测试点1和3的问题在于：判断异性缘的过程中，如果两个人之间没有输入距离(即无穷)，那么这种情况也是不能舍掉的。因为这也属于异性缘的一种，只是这个人一个异性缘都没有罢了。

做法：

1.初始化图

2.记录每个人的距离

3.floyd算每个人的最小距离

4.求每个人的异性缘

5.输出“大众情人”

代码：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 520,INF = 0x3f3f3f3f;

int g[N][N],sexy[N];//g存每个人之间的距离，sexy每个人的异性缘，我这里没有算除法
char sex[N];//性别
int n;

void init()//初始化图
{
    for(int i = 1;i <= n;i++)
        for(int j = 1;j <= n;j++)
            if(i == j) g[i][j] = 0;
            else g[i][j] = INF;
}
void floyd()//floyd算法
{
    for(int k = 1;k <= n;k++)//注意循环顺序
        for(int i = 1;i <= n;i++)
            for(int j = 1;j <= n;j++)
                g[i][j] = min(g[i][j],g[i][k]+g[k][j]);
}
void print(int v,char s)
{
    int flag = 0;
    for(int i = 1;i <= n;i++)
        if(sex[i] == s && sexy[i] == v)
        {
            if(flag) printf(" ");
            printf("%d",i);
            flag = 1;
        }
    puts("");
}
int main()
{
    scanf("%d",&n);
    init();//初始化
    for(int i = 1;i <= n;i++)//记录信息
    {
        char se[2];
        int k = 0;
        scanf("%s%d",se,&k);
        sex[i] = *se;
        while(k--)
        {
            int a = 0,w = 0;
            scanf("%d:%d",&a,&w);
            g[i][a] = min(w,g[i][a]);
        }
    }
    floyd();//算每个人的最短距离
    for(int i = 1;i <= n;i++)//求每个人的异性缘
        for(int j = 1;j <= n;j++)// 是 j -> i，算i的异性缘
            if(sex[i] != sex[j]) sexy[i] = max(g[j][i],sexy[i]);
    
    int ma = INF,fa = INF;
    for(int i = 1;i <= n;i++)//异性缘最好的
        if(sex[i] == 'F') fa = min(fa,sexy[i]);
        else ma = min(ma,sexy[i]);
    print(fa,'F');
    print(ma,'M');
    return 0;
}