【计蒜课 每周三题】2023-02-25 唱歌

唱歌

题目描述

ame 是一个可爱的女孩子，她想要唱歌。

一共有 \(n\) 首歌，第 \(i\) 首歌的长度 \(a_i\)，同时唱第 \(i\) 首歌的满意值为 \(b_i\)。

ame 喜欢的歌满足 \(a_i\leq m\)，同时有 \(k\) 首歌 \(c_1,c_2,\ldots,c_k\) 是 ame 不喜欢的。

其中 \(1\leq c_i\leq n\)，表示编号为 \(c_i\) 的歌曲 ame 是不喜欢的。

请求出 ame 喜欢的歌中满意值第 \(p\) 大的歌曲的编号，如果不存在输出 aaaaaaaaa。

输入格式

输入共四行。

第一行输入 \(4\) 个正整数 \(n,m,k,p\)。

第二行输入 \(n\) 个正整数 \(a_1,a_2,...,a_n\)。

第三行输入 \(n\) 个正整数 \(b_1,b_2,...,b_n\)。

第四行输入 \(k\) 个正整数 \(c_1,c_2,...,c_k\)。

输出格式

输出共一行，输出满意值第 \(p\) 大的歌曲的编号；若不存在，则输出 aaaaaaaaa。

数据范围

对于 \(40\%\) 的数据，有 \(1\leq n\leq 5,1\leq m\leq 10,1\leq a_i \leq m,k=0\)。

对于另外 \(30\%\) 的数据，有 \(1\leq n\leq 10,0\leq k \leq n,1\leq a_i,b_i,m\leq 100\)。

对于 \(100\%\) 的数据，有 \(1\leq n,p\leq 1000,0\leq k\leq n,1\leq c_i\leq n,1\leq a_i,b_i,m\leq 1000\)，数据保证 \(b_i\) 互不相同。

题解

首先除去不喜欢的歌，将喜欢的歌曲放入结构体中，再对结构体按照满意程度从大到小排序即可。

参考代码

#include <bits/stdc++.h>

using namespace std;
const int N = 1010;
int a[N], b[N], c[N];
/*
测试用例：
5 5 0 1
1 2 3 4 5
1 2 3 4 5

n=5 首歌
m=5 喜欢的歌长度小于5
k=0 没有不喜欢的
p=1 满意值第一大的是哪首歌

输入每首歌的长度a[i]
输入每首歌的喜欢程度值b[i]

答案：
5
*/

struct Node {
    int id;
    int value;
    const bool operator<(const Node &b) const {
        return value > b.value;
    }
} d[N];
int dl;

int main() {
    // 必须符合计蒜客的输入输出要求，否则不能完成提交！
    freopen("song.in", "r", stdin);
    freopen("song.out", "w", stdout);

    // n首歌，m:喜欢歌的长度a[i]<m,k：不喜欢歌的数量,p:满意值第p大
    int n, m, k, p;
    cin >> n >> m >> k >> p;
    for (int i = 1; i <= n; i++) cin >> a[i]; // 歌的长度
    for (int i = 1; i <= n; i++) cin >> b[i]; // 满意值

    // 超过长度m限制不喜欢
    for (int i = 1; i <= n; i++)
        if (a[i] > m) c[i] = 1;

    // 特殊指定的不喜欢
    for (int i = 1; i <= k; i++) {
        int x;
        cin >> x;
        c[x] = 1; // 不喜欢,当桶用
    }

    for (int i = 1; i <= n; i++)
        if (!c[i]) d[dl++] = {i, b[i]};
    sort(d, d + dl);

    if (d[p - 1].value)
        printf("%d\n", d[p - 1].id);
    else
        puts("aaaaaaaaa");

    return 0;
}