QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#729249#9573. Social Mediaucup-team4967#WA 50ms3820kbC++231.3kb2024-11-09 16:46:332024-11-09 16:46:41

Judging History

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

  • [2024-11-09 16:46:41]
  • 评测
  • 测评结果:WA
  • 用时:50ms
  • 内存:3820kb
  • [2024-11-09 16:46:33]
  • 提交

answer

#include <bits/extc++.h>

using ll = long long;

using pll = std::pair<ll, ll>;
using pii = std::pair<int, int>;

template <class T>
bool chmax(T& x, const T& v) {
  if (x < v) {
    x = v;
    return true;
  } else {
    return false;
  }
}

void solve() {
  int N, M, K;
  std::cin >> N >> M >> K;
  std::set<int> friends;
  for (int i = 0; i < N; i++) {
    int f;
    std::cin >> f;
    f--;
    friends.insert(f);
  }

  std::map<pii, int> two;
  std::vector<int> one(K);

  int fans = 0;

  for (int i = 0; i < M; i++) {
    int a, b;
    std::cin >> a >> b;
    a--, b--;

    if (a > b) std::swap(a, b);

    bool ainf = friends.contains(a), binf = friends.contains(b);

    if (ainf && binf) {
      fans++;
    } else if (ainf && !binf) {
      one[b]++;
    } else if (!ainf && binf) {
      one[a]++;
    } else if (a == b) {
      one[a]++;
    } else {
      two[{a, b}]++;
    }
  }

  if (K == 1) {
    std::cout << 0 << std::endl;
    return;
  }

  int nans = 0;

  std::vector tmp = one;

  std::ranges::sort(tmp, std::greater<>());

  nans = tmp[0] + tmp[1];

  for (auto& [p, c] : two) {
    auto [a, b] = p;
    chmax(nans, c + one[a] + one[b]);
  }

  std::cout << fans + nans << std::endl;
}

int main() {
  int T;
  std::cin >> T;
  while (T--) {
    solve();
  }
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

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: 50ms
memory: 3756kb

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
6
5
1
11
19
4
3
10
4
1
4
19
15
2
18
4
17
5
1
2
5
17
3
2
6
15
6
3
6
4
4
7
3
7
4
1
16
15
2
3
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
7
9
6
8
2
17
9
5
5
3
6
6
20
8
13
11
10
5
3
6
3
1
8
5
8
11
7
14
10
9
9
11
7
9
5
2
8
14
10
5
3
5
5
10
1
6
9
16
5
3
19
1
4
8
8
10
4
2
1
15...

result:

wrong answer 153rd numbers differ - expected: '10', found: '0'