Question 1: Rat Count House
Problem Description :
The function accepts two positive integers ‘r’ and ‘unit’ and a positive integer array ‘arr’ of size ‘n’ as its argument ‘r’ represents the number of rats present in an area, ‘unit’ is the amount of food each rat consumes and each ith element of array ‘arr’ represents the amount of food present in ‘i+1’ house number, where 0 <= i
Note:
- Return -1 if the array is null
- Return 0 if the total amount of food from all houses is not sufficient for all the rats.
- Computed values lie within the integer range.
Example:
Input:
- r: 7
- unit: 2
- n: 8
- arr: 2 8 3 5 7 4 1 2
Output:
4
Explanation:
Total amount of food required for all rats = r * unit
= 7 * 2 = 14.
The amount of food in 1st houses = 2+8+3+5 = 18. Since, amount of food in 1st 4 houses is sufficient for all the rats. Thus, output is 4.
C++
#include<bits/stdc++.h>
using namespace std;
int calculate (int r, int unit, int arr[], int n)
{
if (n == 0)
return -1;
int totalFoodRequired = r * unit;
int foodTillNow = 0;
int house = 0;
for (house = 0; house < n; ++house)
{
foodTillNow += arr[house];
if (foodTillNow >= totalFoodRequired)
{
break;
}
}
if (totalFoodRequired > foodTillNow)
return 0;
return house + 1;
}
int main ()
{
int r;
cin >> r;
int unit;
cin >> unit;
int n;
cin >> n;
int arr[n];
for (int i = 0; i < n; ++i)
{
cin >> arr[i];
}
cout << calculate (r, unit, arr, n);
return 0;
}C Code:
#include <stdio.h>
int calculate (int r, int unit, int arr[], int n)
{
if (n == 0)
return -1;
int totalFoodRequired = r * unit;
int foodTillNow = 0;
int house = 0;
for (house = 0; house < n; ++house)
{
foodTillNow += arr[house];
if (foodTillNow >= totalFoodRequired)
{
break;
}
}
if (totalFoodRequired > foodTillNow)
return 0;
return house + 1;
}
int main ()
{
int r;
scanf ("%d", &r);
int unit;
scanf ("%d", &unit);
int n;
scanf ("%d", &n);
int arr[n];
for (int i = 0; i < n; ++i)
{
scanf ("%d", &arr[i]);
}
printf ("%d", calculate (r, unit, arr, n));
return 0;
}Python:
def calculate (r, unit, arr, n):
if n == 0:
return -1
totalFoodRequired = r * unit
foodTillNow = 0
house = 0
for house in range (n):
foodTillNow += arr[house]
if foodTillNow>=totalFoodRequired:
break
if totalFoodRequired > foodTillNow:
return 0
return house + 1
r = int (input ())
unit = int (input ())
n = int (input ())
arr = list (map (int, input ().split ()))
print (calculate (r, unit, arr, n))Java:
import java.util.*;
class Main
{
public static int solve (int r, int unit, int arr[], int n)
{
if (arr == null)
return -1;
int res = r * unit;
int sum = 0;
int count = 0;
for (int i = 0; i < n; i++)
{
sum = sum + arr[i];
count++;
if (sum >= res)
break;
}
if (sum < res)
return 0;
return count;
}
public static void main (String[]args)
{
Scanner sc = new Scanner (System.in);
int r = sc.nextInt ();
int unit = sc.nextInt ();
int n = sc.nextInt ();
int arr[] = new int[n];
for (int i = 0; i < n; i++)
arr[i] = sc.nextInt ();
System.out.println (solve (r, unit, arr, n));
}
}