QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#721200#9573. Social MediaBeater#WA 29ms3612kbC++202.4kb2024-11-07 15:33:402024-11-07 15:33:41

Judging History

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

  • [2024-11-07 15:33:41]
  • 评测
  • 测评结果:WA
  • 用时:29ms
  • 内存:3612kb
  • [2024-11-07 15:33:40]
  • 提交

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'