Skip to content
Programming101
Programming101

Learn everything about programming

  • Home
  • CS Subjects
    • IoT – Internet of Things
    • Digital Communication
    • Human Values
  • Programming Tutorials
    • C Programming
    • Data structures and Algorithms
    • 100+ Java Programs
    • 100+ C Programs
  • HackerRank Solutions
    • HackerRank Algorithms Solutions
    • HackerRank C problems solutions
    • HackerRank C++ problems solutions
    • HackerRank Java problems solutions
    • HackerRank Python problems solutions
Programming101
Programming101

Learn everything about programming

HackerEarth Number of blocks problem solution

YASH PAL, 31 July 2024
In this HackerEarth Number of blocks problem solution You are given N cities and the ith city contains a[i] blocks. If you want to build a road between ith and jth cities (i!=j), then the number of blocks needed is gcd(a[i],a[j]). Here gcd is the greatest common divisor. You have to build roads in such a way that any person can go from any city to another city in exactly one way, not more than one way between two cities. You have to maximize the total number of blocks used to build roads.
HackerEarth Number of blocks problem solution

HackerEarth Number of blocks problem solution.

#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define pb push_back
#define all(x) x.begin(),x.end()

#define pll pair<ll,ll>
#define ff first
#define ss second

#define MOD 1000000007

const int N = 100005;

vector<int> v[N];

struct DSU
{
vector<ll> p,sz;
void intialize(int x)
{
p.resize(x+5);
sz.resize(x+5);
for(ll i=0;i<x;i++) p[i] = i,sz[i] = 1;
}
ll getp(ll k)
{
while(k!=p[k])
{
p[k]=p[p[k]];
k=p[k];
}
return k;
}
ll getsize(ll k)
{
return sz[getp(k)];
}
void unionn(ll u,ll v)
{
ll p1=getp(u),p2=getp(v);
if(p1==p2) return;
if(sz[p1]>sz[p2]) swap(p1,p2);
sz[p2]+=sz[p1];
sz[p1]=0;
p[p1]=p[p2];
}
};

void solve()
{
int n;
cin >> n;
for(int i = 0, x; i < n; i++) {
cin >> x;
v[x].pb(i);
}
DSU dsu;
dsu.intialize(n + 5);

long long ans = 0;

for(int i = N - 1; i >= 1; i--) {

for(int j = 1; j < v[i].size(); j++) {
dsu.unionn(v[i][j], v[i][j - 1]);
ans += i;
}

vector<int> ids;
if(v[i].size() > 0) {
ids.pb(v[i][0]);
}
for(int j = i + i; j < N; j += i) {
if(v[j].size() > 0) {
ids.pb(v[j][0]);
}
}
for(int j = 1; j < ids.size(); j++) {
if(dsu.getp(ids[j]) != dsu.getp(ids[j - 1])) {
ans += i;
dsu.unionn(ids[j], ids[j - 1]);
}
}
}

cout << ans << "n";
}

int main()
{
ios_base::sync_with_stdio(false);
cin.tie(0), cout.tie(0);

int t = 1;
for(int i = 1; i < t + 1; i++) {
solve();
}

}

Second solution

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const int N = 1e5 + 14;

struct Dsu {
int par[N];

Dsu() { memset(par, -1, sizeof par); }

int root(int v) {
return par[v] < 0 ? v : par[v] = root(par[v]);
}

bool fri(int v, int u) {
return root(v) == root(u);
}

bool merge(int v, int u) {
if ((v = root(v)) == (u = root(u)))
return 0;
par[u] += par[v];
par[v] = u;
return 1;
}
} dsu;

int t, n;
vector<int> have[N];

int main() {
ios::sync_with_stdio(0), cin.tie(0);
cin >> n;
for (int i = 0; i < n; ++i) {
int x;
cin >> x;
have[x].push_back(i);
}
ll ans = 0;
for (int w = N - 1; w >= 1; --w) {
int r = -1;
for (int i = w; i < N; i += w)
for(auto x : have[i])
if(r == -1)
r = x;
else if(dsu.merge(r, x))
ans += w;
}
cout << ans << 'n';
}
coding problems solutions

Post navigation

Previous post
Next post
  • HackerRank Separate the Numbers solution
  • How AI Is Revolutionizing Personalized Learning in Schools
  • GTA 5 is the Game of the Year for 2024 and 2025
  • Hackerrank Day 5 loops 30 days of code solution
  • Hackerrank Day 6 Lets Review 30 days of code solution
How to download udemy paid courses for free

Pages

  • About US
  • Contact US
  • Privacy Policy

Programing Practice

  • C Programs
  • java Programs

HackerRank Solutions

  • C
  • C++
  • Java
  • Python
  • Algorithm

Other

  • Leetcode Solutions
  • Interview Preparation

Programming Tutorials

  • DSA
  • C

CS Subjects

  • Digital Communication
  • Human Values
  • Internet Of Things
©2025 Programming101 | WordPress Theme by SuperbThemes