LightOJ_1263.cpp

#include <bits/stdc++.h>
using namespace std;
#define F first
#define S second
#define PB push_back
#define PF push_front
#define P push
#define INC(i,a,b) for (ll i = a; i <= b; i++)
#define DEC(i,b,a) for (ll i = b; i >= a ; i--)
#define inf LLONG_MAX
#define neginf LLONG_MIN
#define mod  1000000007
#define eps 1e-9
typedef ostringstream OS ;
typedef stringstream SS ;
typedef long long ll ;
typedef unsigned long long ull;
typedef pair < ll , ll > PLL ;
typedef pair < char,ll > PCL ;
typedef deque < double > DD ;
typedef deque < PCL > DCL ;
typedef deque < ll > DL ;
typedef deque < PLL > DLL ;
typedef deque < char > DC ;
typedef deque < string > DS ;
typedef vector < double > VD;
typedef vector < PCL > VCL ;
typedef vector < ll > VL;
typedef vector < PLL > VLL ;
typedef vector < char > VC ;
typedef vector < string > VS ;
typedef map < ll ,ll > MLL ;
typedef map < char,ll > MCL;
typedef map < ll,char > MLC;
typedef map < string,ll> MSL;
typedef priority_queue < PLL > PQLL ;
typedef priority_queue < ll > PQL ;
typedef stack < ll > SKL ;
typedef stack < PLL > SKLL ;
typedef queue < ll > QL ;
typedef queue < PLL > QLL ;
typedef set < ll > SL ;
typedef set < PLL > SLL ;
typedef set < char > SC ;
 
string numtostr(ll n) {
     OS str1 ;
     str1 << n ;
     return str1.str();
}
ll strtonum(string s) {
     ll x ;
     SS str1(s);
     str1 >> x ;
     return x ;
}
ll GCD(ll a, ll b) {
    if ( b == 0 ) return a ;
    else return GCD(b,a%b);
}
ll LCM(ll a , ll b) {
    ll gcd = GCD(a,b);
     return (a/gcd)*b ;
}
ll v,e,total,ccsize ;
vector < VL > graph ;
VL visited,capa ;
void setting() {
    graph.clear();
    visited.clear();
    capa.clear();
    visited.resize(v+5,0);
    graph.resize(v+5);
    capa.resize(v+5,0);
}
void dfs(ll node) {
    visited[node] = 1 ;
    total += capa[node];
    ccsize++ ;
    for ( ll child : graph[node]) {
        if (!visited[child]) {
            dfs(child);
        }
    }

}
 
int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    ll t,c = 0 ;
    scanf("%lld",&t);
    while(t--) {
        scanf("%lld %lld",&v,&e);
        setting();    
        INC(i,1,v) scanf("%lld",&capa[i]);
        INC(i,1,e) {
            ll node1,node2 ;
            scanf("%lld %lld",&node1,&node2);
            graph[node1].PB(node2);
            graph[node2].PB(node1);
        }
        bool fact = true ;
        ll prevres = -1 ;
        INC(i,1,v) {
            if(!visited[i]) {
               total = 0 ;
               ccsize = 0 ;
               dfs(i);
               if ( total % ccsize != 0) {
                   fact = false ;
                   break ;
               }
               else {
                   if ( prevres != -1 && total/ccsize != prevres) {
                       fact = false ;
                       break ;
                   }
                   else prevres = total/ccsize ;
               }
            }
        }
        if (fact) printf("Case %lld: Yes\n",++c);
        else printf("Case %lld: No\n",++c);
    }
    return 0 ;
}