Add problem 1899 - Merge Triplets to Form Target Triplet

Author: anirudhology

problems/greedy/merge_triplets_to_form_target_triplet.py

@@ -0,0 +1,21 @@
+from typing import List
+
+
+class MergeTripletsToFormTargetTriplet:
+    @staticmethod
+    def mergeTriplets(triplets: List[List[int]], target: List[int]) -> bool:
+        valid_indices = []
+        for i in range(len(triplets)):
+            if triplets[i][0] <= target[0] and triplets[i][1] <= target[1] and triplets[i][2] <= target[2]:
+                valid_indices.append(i)
+
+        is_x_present, is_y_present, is_z_present = False, False, False
+        for index in valid_indices:
+            if not is_x_present and triplets[index][0] == target[0]:
+                is_x_present = True
+            if not is_y_present and triplets[index][1] == target[1]:
+                is_y_present = True
+            if not is_z_present and triplets[index][2] == target[2]:
+                is_z_present = True
+
+        return is_x_present and is_y_present and is_z_present

tests/greedy/merge_triplets_to_form_target_triplet_test.py

@@ -0,0 +1,32 @@
+import unittest
+
+from problems.greedy.merge_triplets_to_form_target_triplet import MergeTripletsToFormTargetTriplet
+
+
+class TestMergeTripletsToFormTargetTriplet(unittest.TestCase):
+
+    def setUp(self):
+        self.merge_triplets = MergeTripletsToFormTargetTriplet()
+
+    def test_merge_triplets_all_present(self):
+        self.assertTrue(self.merge_triplets.mergeTriplets([[2, 5, 3], [1, 8, 4], [1, 7, 5]], [2, 7, 5]))
+
+    def test_merge_triplets_not_all_present(self):
+        self.assertFalse(self.merge_triplets.mergeTriplets([[2, 3, 4], [1, 2, 3], [1, 2, 3]], [3, 2, 5]))
+
+    def test_merge_triplets_single_triplet_match(self):
+        self.assertTrue(self.merge_triplets.mergeTriplets([[2, 5, 3]], [2, 5, 3]))
+
+    def test_merge_triplets_empty_triplets(self):
+        self.assertFalse(self.merge_triplets.mergeTriplets([], [1, 2, 3]))
+
+    def test_merge_triplets_large_values(self):
+        self.assertTrue(self.merge_triplets.mergeTriplets([[100000, 100000, 100000], [99999, 99999, 99999]],
+                                                          [100000, 100000, 100000]))
+
+    def test_merge_triplets_zero_target(self):
+        self.assertTrue(self.merge_triplets.mergeTriplets([[0, 0, 0], [1, 1, 1], [0, 1, 0]], [0, 0, 0]))
+
+
+if __name__ == '__main__':
+    unittest.main()