QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#721200 | #9573. Social Media | Beater# | WA | 29ms | 3612kb | C++20 | 2.4kb | 2024-11-07 15:33:40 | 2024-11-07 15:33:41 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define endl '\n'
void solve(){
int n,m,k;
cin>>n>>m>>k;
int f[n+1];
vector<int > vis(k+1,0);
for(int i=1;i<=n;i++){
cin>>f[i];
vis[f[i]]=1;
}
vector<vector<int > > edge(k+1);
for(int i=1;i<=m;i++){
int u,v;
cin>>u>>v;
if(u!=v){
edge[u].push_back(v);
}
edge[v].push_back(u);
}
ll ans=0;
for(int i=1;i<=k;i++){
for(auto t:edge[i]){
if(vis[i]==1&&vis[t]==1){
ans++;
}
}
}
ans/=2;
// cout<<ans<<endl;
vector<int > val(k+1,0);
for(int i=1;i<=k;i++){
if(vis[i]==1){
continue;
}
for(auto t:edge[i]){
if(vis[t]==1){
val[i]++;
}else if(t==i){
val[i]++;
}
}
}
vector<map<int ,int > > mp(k+1);
// vector<priority_queue<int ,vector<int > ,greater<int > > > Q(k+1);
for(int i=1;i<=k;i++){
if(vis[i]==1){
continue;
}
for(auto t:edge[i]){
if(vis[t]==1){
continue;
}
if(t==i){
continue;
}
mp[i][t]++;
}
}
int res=0;
vector<int > mx;
for(int i=1;i<=k;i++){
if(vis[i]==1){
continue;
}
mx.push_back(val[i]);
for(auto t:edge[i]){
if(vis[t]==1){
continue;
}
if(t==i){
continue;
}
int st=val[t]+mp[i][t]+val[i];
// cout<<st<<" ";
// cout<<i<<" "<<t<<" "<<val[i]<<" "<<val[t]<<endl;
res=max(res,st);
}
// if(!Q[i].empty()){
// int st=val[i]+Q[i].top();
// res=max(res,st);
// }else{
// res=max(res,val[i]);
// }
}
sort(mx.begin(),mx.end(),greater<int > ());
int sum=0;
for(int i=0;i<min((int)mx.size(),2);i++){
sum+=mx[i];
}
res=max(res,sum);
ans+=res;
cout<<ans<<endl;
}
int main (){
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
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: 3492kb
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: 29ms
memory: 3612kb
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 9 19 4 3 10 4 1 4 18 15 2 18 4 17 4 1 2 5 15 3 1 6 15 5 3 6 4 3 7 3 7 4 1 15 14 2 3 6 11 11 7 5 7 8 5 7 11 16 1 4 9 8 13 2 5 17 16 15 8 19 13 7 11 6 9 6 7 2 16 8 5 5 3 6 6 20 8 13 10 9 5 2 6 3 1 8 4 8 10 7 13 9 9 8 11 6 9 4 2 8 13 9 5 3 5 5 9 1 6 9 15 5 3 18 1 3 8 7 10 4 1 1 13 1 9 ...
result:
wrong answer 8th numbers differ - expected: '11', found: '9'