LeetCode 3870 & 3871: from a simple to a general solution

LeetCode problems 3870 and 3871 clearly show the transition from a simple case to a generalized one...