Python Code Counting Rock Samples Problem (TCS CodeVita) | PrepInsta

July 2, 2020

Python Code for Counting Rock Sample Problem

Counting Rock Samples

Counting Rock Samples is one of the sample problem of TCS CodeVita Season 9 coding competition. This is a global level competition organized by TCS every year, from last 8 years. Every year more than 200k coders participate in this coding competition for the title of World’s best Coder.

All TCS CodeVita Questions with Answers

Problem Description

Question – : Juan Marquinho is a geologist and he needs to count rock samples in order to send it to a chemical laboratory. He has a problem: The laboratory only accepts rock samples by a range of its size in ppm (parts per million).

Juan Marquinho receives the rock samples one by one and he classifies the rock samples according to the range of the laboratory. This process is very hard because the number of rock samples may be in millions.

Juan Marquinho needs your help, your task is to develop a program to get the number of rocks in each of the ranges accepted by the laboratory.

Input Format: An positive integer S (the number of rock samples) separated by a blank space, and a positive integer R (the number of ranges of the laboratory); A list of the sizes of S samples (in ppm), as positive integers separated by space R lines where the ith line containing two positive integers, space separated, indicating the minimum size and maximum size respectively of the ith range.

Output Format: R lines where the ith line contains a single non-negative integer indicating the number of the samples which lie in the ith range.

Constraints:

  • 10 <= S <= 10000
  • 1 <= R <= 1000000
  • 1<=size of Sample <= 1000

Example 1

  • Input: 10 2
  • 345 604 321 433 704 470 808 718 517 811
  • 300 350
  • 400 700

Output: 2 4

Explanation:

There are 10 samples (S) and 2 ranges ( R ). The samples are 345, 604,811. The ranges are 300-350 and 400-700. There are 2 samples in the first range (345 and 321) and 4 samples in the second range (604, 433, 470, 517). Hence the two lines of the output are 2 and 4

Example 2

  • Input: 20 3
  • 921 107 270 631 926 543 589 520 595 93 873 424 759 537 458 614 725 842 575 195
  • 1 100
  • 50 600
  • 1 1000

Output: 1 12 20

Explanation:

There are 20 samples and 3 ranges. The samples are 921, 107 195. The ranges are 1-100, 50-600 and 1-1000. Note that the ranges are overlapping. The number of samples in each of the three ranges are 1, 12 and 20 respectively. Hence the three lines of the output are 1, 12 and 20.

Python Code

samples, ranges =[int(i) for i in input().split()]
count = 0
final = []
arr = list(map(int, input().split()))
for i in range(0, ranges):
    range1, range2 = [int(i) for i in input().split()]
    for j in range(0, samples):
        if range1 <= arr[j] <= range2:
            count = count + 1
    final.append(count)
    count = 0
for i in range(0, len(final)):
    print(final[i], end=" ")
Output
10 2
345 604 321 433 704 470 808 718 517 811
300 350
400 700
2 4

Counting Rock Samples in Other Coding Languages

C

To find the solution of Counting Rock Sample Problem in C Programming language click on the button below:

C

C++

To find the solution of Counting Rock Sample Problem in C++ Programming language click on the button below:

C++

Java

To find the solution of Counting Rock Sample Problem in Java Programming language click on the button below:

Java

18 comments on “Python Code Counting Rock Samples Problem (TCS CodeVita) | PrepInsta”


  • Ravi

    class Lab:
    def __init__(self,sample,ranges):
    self.s = sample
    self.r = ranges
    self.samplelist = []
    self.allRange = []
    self.getData()
    self.checkData()
    def getData(self):
    for i in range(self.s):
    self.samples = int(input(“Sample “+str(i+1)+” “))
    self.samplelist.append(self.samples)
    for i in range(self.r):
    self.rl = []
    print(“Give Range “+str(i+1))
    self.rStart = int(input(“Range Start : “))
    self.rEnd = int(input(“Range End : “))
    self.rl.append(self.rStart)
    self.rl.append(self.rEnd)
    self.allRange.append(self.rl)
    #print(self.allRange)

    def checkData(self):
    cnt = 0
    for i in range(self.r):
    for j in self.samplelist:
    if j >=self.allRange[i][0] and j<=self.allRange[i][1]:
    cnt += 1
    print("There are "+str(cnt)+" samples in range "+str(self.allRange[i][0])+" to "+str(self.allRange[i][1]))
    cnt = 0
    lab1 = Lab(10,2)

1 2