Python Code for Worker (TCS Codevita) | PrepInsta
Worker
TCS CodeVita is a coding competition organized by TCS every year, in search of world’s best coder. This is a global level coding competition in which coders from all around the world compete for the title of World’s Best Coder. Worker is one of the sample problem of this year TCS CodeVita season 11 competition.
Problem Description
Question -: This is a resource to task allocation problem. There are two types of tasks viz. one that the human needs to do and the other which as amenable to machine processing. A human-oriented task cannot be performed by machine and vice-versa. Also, assume that all human workers are equally efficient. So, the same task will be performed in the same time, irrespective of which human worker does that task.
Given the number of workers, number of tasks and time taken to finish each task, compute which task was finished at what time and which worker performed that task. Some rules and conventions that need to be adhered to while doing task allocation to workers are as below.
- Machine processing for any task takes 0 minutes
- Time taken (in minutes) by a human worker to complete a particular task is mentioned in the input
- Workers and tasks are identified by their suffixes i.e., Worker 1 is W1 and Task 1 is T1.
- Task information is provided in the input in form of a tuple that consists of 3 things viz. <Taskname, TaskType, CompletionTime> where
- Taskname will be of form T1, T2, T3 etc.
- TaskType will be a keyword which can be either {Manual, Machine}
- CompletionTime is a number denoting time in minutes needed for manual tasks and 0 for machine-bound tasks
- A worker will still need to be assigned even if a task can be done only by the machine
- Tasks need to be processed in sequential order i.e., T1 has to be allocated before T2
- Initially when all workers are free, worker Wi will be allocated task before worker Wi+1 i.e., in general a worker with lower suffix will be assigned to the task before worker with a higher suffix.
- If one or more workers are available at the same time, then the worker who processed lower suffix task will get the next task allocated prior to the other workers.
- Assume the switching time from one task to another as 0 min
Refer Examples section for better understanding of how tasks are allocated to workers
Compute which task was completed at what time and which worker performed that task. The output specifications provide information on how output should be printed.
Compute which task was completed at what time and which worker performed that task. The output specifications provide information on how output should be printed.
Constraints :
- 1 <= N <= 10
- 0 < M <= 100
Input :
First line contains an integer N which denotes the number of workers
Second line contains an integer M which denotes the number of tasks
Next M lines contain a task information tuple as described in the previous section.
First line contains an integer N which denotes the number of workers
Second line contains an integer M which denotes the number of tasks
Next M lines contain a task information tuple as described in the previous section.
Output :
Output should adhere to the following specifications.
Output should adhere to the following specifications.
- Each output should be a tuple comprising of 3 pieces of information viz <Worker name, task name, time at which the task was completed>
- The tuple should be first sorted in ascending order of completion time and then in ascending order of task name
For better understanding refer to the Examples section.
Time Limit : 1 secs
Example 1 :
Input :
3
4
T1 Machine
T2 Manual 5
T3 Machine
T4 Machine
Output :
W1 T1 0
W3 T3 0
W1 T4 0
W2 T2 5
Explanation :
Here, N=3 denotes number of workers W1, W2, W3.
First, Workers W1, W2, W3 will be assigned with tasks T1, T2, T3 at the 0th minute, respectively.
Here, T1 is Machine-bound so it will be completed in 0th minute and then W1 will be available for the remaining tasks. T2 is Manual and it will be completed in 5th minute. So, W2 will be free after 5 minutes. T3 again is Machine-bound so it will be completed in 0th minute and then W3 will be available for remaining tasks.
Now, Workers available for pending tasks in 0th minute = {W1, W3}
Here, W1 will be assigned with next pending task first, because the worker W1 completed task with lower suffix (T1) than W3 (T3).
W1 will be assigned the next task T4 at 0th minute which is a Machine-bound task which will also be completed in 0th minute.
As the number of pending tasks are zero, no further assignment takes place.
W2 will complete T2 at the end of 5th minute. Hence,
First line of output is “W1 T1 0”, where T1 is the task completed by Worker W1 0th minute.
Second line of output is “W3 T3 0”, where T3 is the task completed by Worker W3 in 0th minute.
Third line of output is “W1 T4 0”, where T4 is the task completed by worker W1 in 0th minute.
Fourth line of output is “W2 T2 5”, where T2 is the task completed by W2 in 5th minute.
So, the output is as follows:
W1 T1 0
W3 T3 0
W1 T4 0
W2 T2 5
Example 2 :
Input :
3
4
T1 Machine
T2 Machine
T3 Machine
T4 Machine
Output :
W1 T1 0
W2 T2 0
W3 T3 0
W1 T4 0
Explanation :
Given N = 3 denotes number of workers W1, W2 and W3
First, Workers W1, W2, W3 will be assigned with tasks T1, T2, T3 in 0th minute, respectively.
Here, T1, T2 and T3 are Machine-bound and will be completed in 0th minute, after completion of these tasks W1, W2, W3 will be available for the remaining tasks.
Since W1 is the first worker in the list of available workers at 0th minute. W1 will pick up the 4th task and will complete it in 0th minute, because it is also a machine bounded task.
Hence the output is as follows:
W1 T1 0
W2 T2 0
W3 T3 0
W1 T4 0
Python
Python
Run
def main():
n = int(input())
m = int(input())
tasks = []
workers = []
for i in range(m):
task_name, task_type, completion_time = input().split()
tasks.append({"name": task_name, "type": task_type, "completion_time": int(completion_time)})
tasks.sort(key=lambda x: x["name"])
for i in range(n):
worker_name = f"W{i+1}"
workers.append({"name": worker_name, "time_available": 0})
time_taken = 0
i = 0
j = 0
while i < m or j < n:
if i < m and tasks[i]["completion_time"] == 0:
worker_name = workers[j]["name"]
print(f"{worker_name} {tasks[i]['name']} {time_taken}")
i += 1
j += 1
elif i < m and tasks[i]["completion_time"] > 0:
min_index = -1
for k in range(n):
if workers[k]["time_available"] <= time_taken:
min_index = k
break
if min_index != -1:
worker_name = workers[min_index]["name"]
print(f"{worker_name} {tasks[i]['name']} {time_taken + tasks[i]['completion_time']}")
workers[min_index]["time_available"] = time_taken + tasks[i]["completion_time"]
i += 1
else:
min_index = -1
for k in range(n):
if workers[k]["time_available"] <= time_taken:
min_index = k
break
if min_index != -1:
time_taken = workers[min_index]["time_available"]
j = min_index + 1
if __name__ == "__main__":
main()
