Skip to content
Programmingoneonone
Programmingoneonone
  • Home
  • CS Subjects
    • Internet of Things (IoT)
    • 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
  • Work with US
Programmingoneonone
Programmingoneonone

HackerEarth Water supply problem solution

YASH PAL, 31 July 2024
In this HackerEarth Water supply problem solution Your country can be represented by N cities connected using N – 1 roads. There exists an edge between city (i, i + 1) for all i from 1 to n – 1.
You have to set up a connection for water supply. You set this in one city and water gets transported from it to other cities using road transport. Certain cities are blocked which means that water cannot pass through this city. Determine the maximum number of cities to which water can be supplied.
HackerEarth Water supply problem solution

HackerEarth Water supply problem solution.

#include<bits/stdc++.h>
#define ll long long int
using namespace std;
vector<int>adjacent[200005];
bool visited[200005];
bool blocked[200005];
int sz;

void dfs(int u)
{
visited[u] = true;
sz++;
for(int i = 0 ; i < adjacent[u].size() ; i++)
{
int v = adjacent[u][i];
if(!visited[v] && blocked[v] == 0)
{
dfs(v);
}
else if(blocked[v])
{
sz++;
}
}
}


int main(){
int n;
cin>>n;

for(int i = 0 ; i < n - 1 ; i++)
{
int u,v;
cin>>u>>v;
adjacent[u].push_back(v);
adjacent[v].push_back(u);
}

for(int i = 1 ; i <= n ; i++)
{
int u;
cin>>u;

blocked[i] = u;
}

int ans = 1;
for(int i = 1 ; i <= n ; i++)
{
if(visited[i] == false && blocked[i] == 0)
{
sz = 0;
dfs(i);
ans = max(ans,sz);
}
}

cout<<ans<<endl;

}
coding problems solutions

Post navigation

Previous post
Next post

Related website

The Computer Science

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
  • YouTube
  • LinkedIn
  • Facebook
  • Pinterest
  • Instagram
©2025 Programmingoneonone | WordPress Theme by SuperbThemes