PAT L2-019 悄悄关注

https://pintia.cn/problem-sets/994805046380707840/problems/994805059731177472

新浪微博上有个“悄悄关注”，一个用户悄悄关注的人，不出现在这个用户的关注列表上，但系统会推送其悄悄关注的人发表的微博给该用户。现在我们来做一回网络侦探，根据某人的关注列表和其对其他用户的点赞情况，扒出有可能被其悄悄关注的人。

输入格式：

输入首先在第一行给出某用户的关注列表，格式如下：

人数N 用户1 用户2 …… 用户N

其中N是不超过5000的正整数，每个用户i（i=1, ..., N）是被其关注的用户的ID，是长度为4位的由数字和英文字母组成的字符串，各项间以空格分隔。

之后给出该用户点赞的信息：首先给出一个不超过10000的正整数M，随后M行，每行给出一个被其点赞的用户ID和对该用户的点赞次数（不超过1000），以空格分隔。注意：用户ID是一个用户的唯一身份标识。题目保证在关注列表中没有重复用户，在点赞信息中也没有重复用户。

输出格式：

我们认为被该用户点赞次数大于其点赞平均数、且不在其关注列表上的人，很可能是其悄悄关注的人。根据这个假设，请你按用户ID字母序的升序输出可能是其悄悄关注的人，每行1个ID。如果其实并没有这样的人，则输出“Bing Mei You”。

输入样例1：

10 GAO3 Magi Zha1 Sen1 Quan FaMK LSum Eins FatM LLao
8
Magi 50
Pota 30
LLao 3
Ammy 48
Dave 15
GAO3 31
Zoro 1
Cath 60

输出样例1：

Ammy
Cath
Pota

输入样例2：

11 GAO3 Magi Zha1 Sen1 Quan FaMK LSum Eins FatM LLao Pota
7
Magi 50
Pota 30
LLao 48
Ammy 3
Dave 15
GAO3 31
Zoro 29

输出样例2：

Bing Mei You

时间复杂度：$O(logM)$

代码：

#include <bits/stdc++.h>
using namespace std;

int N, M, sum;
double ave;
char s[5010][10];
map<string, int> mp;

struct User {
    char name[10];
    int score;
}user[10010];

struct Node {
    char ID[10];
}node[10010];

bool cmp(const Node& a, const Node& b) {
    return strcmp(a.ID, b.ID) < 0;
}

int main() {
    scanf("%d", &N);
    mp.clear();
    for(int i = 1;i <= N; i ++) {
        scanf("%s", s[i]);
        mp[s[i]] = 1;
    }

    int flag[10010];
    scanf("%d", &M);
    memset(flag, 0, sizeof(flag));
    for(int i = 1; i <= M; i ++) {
        scanf("%s%d", user[i].name, &user[i].score);
        sum += user[i].score;
        if(mp[user[i].name] == 0) flag[i] = 1;
    }
    ave = 1.0 * sum / M;

    int num = 0;
    for(int i = 1; i <= M; i ++) {
        if(flag[i] == 1 && user[i].score >= ave) {
            num ++;
            strcpy(node[num].ID, user[i].name);
        }
    }

    if(num == 0)
        printf("Bing Mei You\n");
    else {
        sort(node + 1, node + 1 + num, cmp);
        for(int i = 1; i <= num; i ++)
            printf("%s\n", node[i].ID);
    }
    return 0;
}