Ice cream

input/code.cpp: In function 'int main()':
input/code.cpp:26:9: warning: unused variable 'vv' [-Wunused-variable]
     int vv = 0,vi=0;
         ^

#include <vector>
#include <set>
#include <iostream>
#include <queue>
using namespace std;
int main(){
  int n,m;
  ios::sync_with_stdio(0);
  cin.tie(0);
  cin>>n>>m;
  vector<int> c;
  c.resize(n);
  for(auto& it : c)cin>>it;
  vector<vector<int>> e;
  e.resize(n);
  for(int i=0;i<m;++i){
    int a,b;
    cin>>a>>b;
    e[a-1].push_back(b-1);
    e[b-1].push_back(a-1);
  }
  vector<int> v;
  v.insert(v.begin(),n,-1);
  for(int i=0;i<n;++i){
    if(v[i]!=-1)continue;
    int vv = 0,vi=0;
    set<int> qq;
    set<int> kk;
    queue<int> q;
    q.push(i);
    v[i]=i;
    while(q.size()){
      vi = q.front();q.pop();
      qq.insert(c[vi]);
      kk.insert(vi);
      for(auto it : e[vi]){
        if(v[it]==-1){v[it]=i;q.push(it);}
      }
      vi++;
    }
    for(auto it : kk)v[it]=qq.size();
  }
  for(auto it : v)cout<<it<<(" \n"[!--n]);
  return 0;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED

Test 14

Verdict: ACCEPTED

Test 15

Verdict: ACCEPTED

Test 16

Verdict: ACCEPTED

Test 17

Verdict: ACCEPTED

Test 18

Verdict: ACCEPTED

Test 19

Verdict: ACCEPTED

Test 20

Verdict: ACCEPTED