আমাদের তিনটি সংখ্যা আছে i, j, k এবং আরেকটি সংখ্যা n। আমাদের সব ট্রিপলেট (i, j, k) বের করতে হবে যাদের জন্য i+j+k != n। এই সমস্যাটি লিস্ট কম্প্রিহেনশন দিয়ে সমাধান করতে হবে।
উদাহরণস্বরূপ, ইনপুট i = 1, j = 1, k = 2, n = 3 হলে আউটপুট হবে [[0, 0, 0], [0, 0, 1], [0, 0, 2], [0, 1, 0], [0, 1, 1], [1, 0, 0], [1, 0, 1], [1, 1, 0], [1, 1, 2]]।
সমাধানের ধাপসমূহ
arr = [[x, y, z] for x in range(i+1) for y in range(j+1) for z in range(k+1) if x+y+z != n]
উদাহরণ
নিচের ইমপ্লিমেন্টেশনটি দেখুন ভালোভাবে বুঝার জন্য:
def solve(i, j, k, n):
arr = [[x, y, z] for x in range(i+1) for y in range(j+1) for z in range(k+1) if x+y+z != n]
return arr
i = 1
j = 1
k = 2
n = 3
print(solve(i, j, k, n)) ইনপুট
1, 1, 2, 3
আউটপুট
[[0, 0, 0], [0, 0, 1], [0, 0, 2], [0, 1, 0], [0, 1, 1], [1, 0, 0], [1, 0, 1], [1, 1, 0], [1, 1, 2]]