2018-2019 ACM-ICPC, Asia East Continent Final L Eventual … Journey

#include<iostream>
using namespace std;
const int MAX=2e5+;
int cnt[MAX];
int ans[MAX];
int a[MAX];
int main() {
    int n,m,zero=,one=;
    cin>>n>>m;
    for(int i=; i<=n; i++) {
        scanf("%d",&a[i]);
        if(a[i]==)zero++;
        else one++;
    }
    while(m--) {
        int x,y;
        scanf("%d%d",&x,&y);
        if(a[x]!=a[y]) {
            cnt[x]++;//记录每个点的桥
            cnt[y]++;
        }
    }
    int A=,B=;
    for(int i=; i<=n; i++) {
        if(a[i]==&&cnt[i])A++;//0类，而且有桥，
        if(a[i]==&&cnt[i])B++;//1类，数量
    }
    for(int i=; i<=n; i++) {
        ans[i]+=cnt[i];//与异类相连的桥的数目，也就是步数
        if(a[i]==)ans[i]+=zero-;//加同类
        else ans[i]+=one-;//同类
        if(cnt[i]==) {//如果 这个站没有建桥
            if(a[i]==) {//如果是0类
                ans[i]+=*B;//一个点只算建在上面的一座桥，其他为在一个点上重复建桥的
                ans[i]+=*(one-B);//
            } else {
                ans[i]+=*A;
                ans[i]+=*(zero-A);
            }
        } else {//如果这个点建桥了
            if(a[i]==)ans[i]+=*(one-cnt[i]);
            //如果是0类， 那么到1类的其他点的距离就是2
            else ans[i]+=*(zero-cnt[i]);
        }
    }
    for(int i=; i<=n; i++)
        cout<<ans[i]<<" ";
    return ;
}