QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#732186#9573. Social Media333zhan#WA 18ms3832kbC++201.5kb2024-11-10 13:37:342024-11-10 13:37:35

Judging History

你现在查看的是最新测评结果

  • [2024-11-10 13:37:35]
  • 评测
  • 测评结果:WA
  • 用时:18ms
  • 内存:3832kb
  • [2024-11-10 13:37:34]
  • 提交

answer

#include <bits/stdc++.h>
#define int long long

using namespace std;

void solve () {
    int n, m, k;
    cin >> n >> m >> k;

    vector <bool> f (k + 1);
    for (int i = 1; i <= n; i ++) {
        int x;
        cin >> x;

        f[x] = true;
    }

    int sum = 0;
    vector <map <int, int>> mp (k + 1);
    vector <int> a (k + 1);
    for (int i = 0; i < m; i ++) {
        int x, y;
        cin >> x >> y;

        if (f[x] && f[y]) {
            sum ++;
            continue;
        }
        if (! f[x] && ! f[y]) {
            mp[x][y] ++;
            mp[y][x] ++;
            continue;
        }
        if (! f[x]) {
            swap (x, y);
        }
        a[y] ++;
    }   

    vector <int> p (k + 1);
    iota (p.begin (), p.end (), 0);
    sort (p.begin () + 1, p.end (), 
        [&](const int &x, const int &y) {
            return a[x] > a[y];
        }
    );

    if (n >= k - 2) {
        cout << m << '\n';
        return;
    }

    int ans = sum;
    for (int i = 1; i <= k; i ++) {
        if (f[i]) {
            continue;
        }
        int tot = a[i] + (p[1] == i ? a[p[2]] : a[p[1]]);
        for (auto [x, y] : mp[i]) {
            tot = max (tot, a[i] + y + a[x]);
        }
        ans = max (ans, sum + tot);
    }

    cout << ans << '\n';
}

signed main () {
    ios::sync_with_stdio (false);
    cin.tie (nullptr);

    int T = 1;
    cin >> T;

    while (T --) {
        solve ();
    }

    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3832kb

input:

5
4 12 7
5 7 3 6
3 6
2 2
1 4
2 4
1 3
7 6
4 1
5 4
1 1
1 1
2 1
3 7
2 7 6
2 4
1 2
3 2
2 5
5 4
2 6
4 6
2 6
1 1 2
1
1 2
2 1 2
1 2
1 2
2 1 100
24 11
11 24

output:

9
5
1
1
1

result:

ok 5 number(s): "9 5 1 1 1"

Test #2:

score: -100
Wrong Answer
time: 18ms
memory: 3584kb

input:

10000
19 12 20
8 12 1 5 11 7 17 13 19 6 3 9 10 15 14 20 4 18 16
4 11
7 1
8 4
16 19
1 13
15 2
16 2
8 7
3 15
11 13
5 20
18 14
17 14 20
2 9 1 12 8 11 10 17 18 16 3 15 5 14 20 13 7
15 10
3 2
5 16
7 8
6 1
6 4
18 16
1 8
4 1
20 6
6 9
4 15
7 5
14 9
1 3 18
9
15 18
17 15
11 14
7 19 7
3 1 2 5 6 4 7
5 1
4 5
3 1...

output:

12
14
1
19
7
5
1
11
19
4
3
10
6
2
4
19
15
2
18
4
17
5
1
2
5
17
3
2
6
15
6
2
6
4
4
7
3
7
4
1
19
15
2
4
6
12
12
7
6
8
8
6
8
11
16
1
4
9
8
14
2
6
19
19
16
8
20
14
8
12
6
9
6
8
2
17
9
5
5
3
5
5
20
8
13
11
10
5
3
7
3
1
8
5
8
11
7
14
10
9
8
11
7
9
5
2
8
14
10
5
3
5
4
10
1
6
10
16
5
3
19
1
4
8
7
10
4
2
1
1...

result:

wrong answer 5th numbers differ - expected: '6', found: '7'