Advertisement
1995. Count Special Quadruplets
UnknownView on LeetCode
Time: O(n²)
Space: O(n)
Approach
Two-loop with HashMap tracking nums[d]-nums[c] values; count matching a+b+c == d.
1995.cs
C#
// Approach: Two-loop with HashMap tracking nums[d]-nums[c] values; count matching a+b+c == d.
// Time: O(n²) Space: O(n)
public class Solution
{
public int CountQuadruplets(int[] nums)
{
int n = nums.Length;
int ans = 0;
Dictionary<int, int> count = new Dictionary<int, int>();
// nums[a] + nums[b] + nums[c] == nums[d]
// => nums[a] + nums[b] == nums[d] - nums[c]
for (int b = n - 1; b > 0; --b)
{ // `b` also represents `c`.
for (int a = b - 1; a >= 0; --a)
ans += count.TryGetValue(nums[a] + nums[b], out int value) ? value : 0;
for (int d = n - 1; d > b; --d)
if (count.ContainsKey(nums[d] - nums[b]))
{
count[nums[d] - nums[b]]++;
}
else
{
count[nums[d] - nums[b]] = 1;
} // b := c
}
return ans;
}
}Advertisement
Was this solution helpful?