每日一题 - 240419 - 24 GPLT 模拟赛 - L1-7 撒狗粮

乐意奥AI

已于 2024-04-21 13:55:46 修改

阅读量347

点赞数 7

分类专栏：每日一题文章标签：算法

于 2024-04-21 13:53:23 首次发布

本文链接：https://blog.csdn.net/CodeToLiao/article/details/137980463

版权

每日一题专栏收录该内容

5 篇文章 0 订阅

订阅专栏

0x00 Link

L1-7 撒狗粮 404
Source - PTA
Tag - 芝士水题 差分
Time - $*$

0x01 Source Code

//
#include <bits/stdc++.h>
using namespace std;
// #define int long long 
#define A pair<string, int>
#define p1 first
#define p2 second

const int N = 1e6 + 7;      // 之后给出一个正整数 M（≤ 80 000），为参加派对的总人数
string s[N];
int cnt[N], sum[N];
bool is_dog[N];
map<string, string> mp;
map<string, int> get_idx;
vector<A> v;
vector<string> ans;

bool cmp(const A& x, const A& y) {
    return x.p2 == y.p2 ? x.p1 < y.p1 : x.p2 > y.p2;    // 如果不唯一，按 ID 递增顺序列出
}

void solve() {
    int n;
    cin >> n;
    while (n--) {
        string s1, s2;
        cin >> s1 >> s2;
        mp[s1] = s2;
        mp[s2] = s1;
    }

    int m;
    cin >> m;
    for (int i = 1; i <= m; i++) {
        cin >> s[i];
        get_idx[s[i]] = i;
    }

    for (int i = 1; i <= m; i++) {
        if (mp.find(s[i]) != mp.end() && get_idx.find(mp[s[i]]) != get_idx.end()) {
            int idx = get_idx[mp[s[i]]];
            if (i < idx) {
                if (i + 1 == idx) {             // 如果一对情侣坐在一起，那么他们两人身边的单身狗就会被撒一脸狗粮
                    int x = i - 1;
                    int y = idx + 1;
                    cnt[x]++, cnt[x + 1]--;
                    cnt[y]++, cnt[y + 1]--;
                } else {                        // 如果他们没有坐在一起，那么所有被夹在他们两人之间的单身狗都会被撒一脸狗粮
                    int x = i + 1;
                    int y = idx - 1;
                    cnt[x]++, cnt[y + 1]--;
                }
            }
        } else {
            is_dog[i] = 1;
        }
    }

    for (int i = 1; i <= m; i++) {
        sum[i] = sum[i - 1] + cnt[i];
        if (is_dog[i]) {
            v.push_back(make_pair(s[i], sum[i]));
        }
    }
    sort(v.begin(), v.end(), cmp);

    ans.push_back(v[0].p1); 
    for (int i = 1; i < v.size(); i++) {
        if (v[i].p2 != v[0].p2) break;      // 找出被撒狗粮最多（以“脸”为单位）的那位单身人士
        ans.push_back(v[i].p1);
    }

    for (int i = 0; i < ans.size(); i++) {
        cout << ans[i] << " \n"[i == ans.size() - 1];   // ID 间用 1 个空格分隔，行的首尾不得有多余空格
    }
}

signed main() {
    int t = 1;
    // scanf("%d", &t);
    while (t--) solve();
    return 0;
}