提出 #52180312

---

ソースコード 拡げる

#include<iostream>
#include<string>
#include<queue>
#include<vector>
#include<cassert>
using namespace std;
typedef long long ll;
const int inf=1<<30;
const ll INF=1LL<<60;
typedef pair<ll,int> P;


int main(){
    int N,M;
    cin>>N>>M;
    vector<ll> A(N);
    for(int i=0;i<N;i++){
        cin>>A[i];
    }
    vector<vector<int>> G(N);
    for(int i=0;i<M;i++){
        int u,v;
        cin>>u>>v;
        u--;v--;
        G[u].push_back(v);
        G[v].push_back(u);
    }

    priority_queue<P,vector<P>,greater<P>> pq;//コストが小さい方から

    vector<ll> cost(N,0);
    for(int i=0;i<N;i++){
        
        for(int to:G[i]){
            cost[i]+=A[to];
        }
        pq.push({cost[i],i});
    }

    vector<bool> is_removed(N,false);
    ll ans=-INF;
    while(!pq.empty()){

        auto [c,now]=pq.top();
        pq.pop();
        if(is_removed[now]) continue;

        is_removed[now]=true;
        //コストcで取り出す
        ans=max(ans,c);

        for(int to:G[now]){
            if(!is_removed[to]){
                //いまだに頂点toは取り出されてない
                cost[to]-=A[now];
                pq.push({cost[to],to});
            }
        }
    }
    cout<<ans<<endl;


}

提出情報

ジャッジ結果