JAVA Program for Collecting Candies (TCS Codevita)

JAVA Program for Collecting Candies (TCS Codevita) | PrepInsta

July 2, 2020

Collecting Candies Problem

TCS annually puts together a coding competition called TCS CodeVita. This competition helps the company to hire some of the best developers aspiring to be a part of the TCS Family. Collecting Candies is one of the sample problem of TCS CodeVita competition. It is solved by using a sorting algorithm with some changes here and there, here we have provided a JAVA solution for this problem.

Problem Description

Question:- Krishna loves candies a lot, so whenever he gets them, he stores them so that he can eat them later whenever he wants to.

He has recently received N boxes of candies each containing Ci candies where Ci represents the total number of candies in the ith box. Krishna wants to store them in a single box. The only constraint is that he can choose any two boxes and store their joint contents in an empty box only. Assume that there are an infinite number of empty boxes available.

At a time he can pick up any two boxes for transferring and if both the boxes contain X and Y number of candies respectively, then it takes him exactly X+Y seconds of time. As he is too eager to collect all of them he has approached you to tell him the minimum time in which all the candies can be collected.

Input Format:

The first line of input is the number of test case T
Each test case is comprised of two inputs
The first input of a test case is the number of boxes N
The second input is N integers delimited by whitespace denoting the number of candies in each box

Output Format: Print minimum time required, in seconds, for each of the test cases. Print each output on a new line.

Constraints:

1 < T < 10
1 < N< 10000
1 < [Candies in each box] < 100009

S. No.	Input	Output
1	1 4 1 2 3 4	19
2	1 5 1 2 3 4	34

Explanation for sample input-output 1:

4 boxes, each containing 1, 2, 3 and 4 candies respectively.Adding 1 + 2 in a new box takes 3 seconds.Adding 3 + 3 in a new box takes 6 seconds.Adding 4 + 6 in a new box takes 10 seconds.Hence total time taken is 19 seconds. There could be other combinations also, but overall time does not go below 19 seconds.

Explanation for sample input-output 2:

5 boxes, each containing 1, 2, 3, 4 and 5 candies respectively.Adding 1 + 2 in a new box takes 3 seconds.Adding 3 + 3 in a new box takes 6 seconds.Adding 4 + 6 in a new box takes 10 seconds.Adding 5 + 10 in a new box takes 15 seconds.Hence total time taken is 34 seconds. There could be other combinations also, but overall time does not go below 33 seconds.

JAVA Code

import java.util.Scanner;
public class Main {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n, i, k = 0, sum = 0, s1 = 0, t, temp = 0, j;
        long c[] = new long[1000009];
        long s[] = new long[100009];
        System.out.println("Enter a no");
        t = sc.nextInt();
        for (int l = 0; l < t; l++)
        {
            n = sc.nextInt();
            for (i = 0; i < n; i++)
                c[i] = sc.nextLong();
            for (i = 0; i < n; i++) {
                for (j = i + 1; j < n; j++) { if (c[i] > c[j])
                    {
                        temp = (int) c[i];
                        c[i] = c[j];
                        c[j] = temp;
                    }
                }
            }
            sum = 0;
            k = 0;
            for (i = 0; i < n; i++)
            {
                sum = (int) (sum + c[i]);
                s[k] = sum;
                k++;
            }
            s1 = 0;
            for (i = 1; i < k; i++)
                s1 = (int) (s1 + s[i]);
            System.out.println(s1);
        }
    }

}

Output
1
4
1 2 3 4

19

Collecting Candies Problem in other Languages

C

To find the solution of Collecting Candies problem in C Programming language click on the button below:

C++

To find the solution of Collecting Candies problem in C++ Programming language click on the button below:

C++

Python

To find the solution of Collecting Candies problem in Python Programming language click on the button below:

Python

One comment on “JAVA Program for Collecting Candies (TCS Codevita) | PrepInsta”

Renganathan
Scanner in =new Scanner(System.in);

int t=in.nextInt();
for(int k=0;k<t;k++)
{

int n=in.nextInt();

int[] count=new int[n-1];

int[] c=new int[n];

for(int i=0;i<n;i++)
{
c[i]=in.nextInt();
}

int sum,l=0;
for(int i=0;i<n-1;i++)
{
if(i==0)
{
sum=c[i]+c[i+1];
count[l++]=sum;

}
else
{
sum=count[l-1]+c[i+1];
count[l++]=sum;

}
sum=0;
}

int sum1=0;
for(int j=0;j<count.length;j++)
{
sum1+=count[j];
}

System.out.println(sum1);

This will work in straightforward

1

Log in to Reply