Подсчет пар с разницей >= К

public class Solution {
    public static int CountPairs(List<int> nums, int k) {
        nums.Sort();
        int n = nums.Count;
        if (k == 0) {
            // формула количества пар i <= j: n * (n + 1) / 2
            return n * (n + 1) / 2;
        }
        int l = 0;
        int r = -1;
        int count = 0;
        while (l < n) {
            // Двигаем правый указатель, пока разница < k
            while (r + 1 < n && nums[r + 1] - nums[l] < k) {
                r++;
            }
            // считаем число пар, дающих в сумме >= k
            // с элементом nums[l] - это все элементы справа
            // от плавающего окна
            count += n - (r + 1);
            l++;
        }
        return count;
    }
}

#include <vector>

using namespace std;

int countPairs(vector<int> nums, int k) {
    sort(nums.begin(), nums.end());
    int n = nums.size();
    if (k == 0) {
        // формула количества пар i <= j: n * (n + 1) // 2
        return n * (n + 1) / 2;
    }
    int l = 0;
    int r = -1;
    int count = 0;
    while (l < n) {
        // Двигаем правый указатель, пока разница < k
        while (r + 1 < n && nums[r + 1] - nums[l] < k) {
            r++;
        }
        // считаем число пар, дающих в сумме >= k
        // с элементом nums[l] - это все элементы справа
        // от плавающего окна
        count += n - (r + 1);
        l++;
    }
    return count;
}

package main

import (
    "sort"
)

func countPairs(nums []int, k int) int {
    n := len(nums)
    if k == 0 {
        return n * (n + 1) / 2 // формула количества пар i <= j
    }
    sort.Ints(nums)
    l, r, count := 0, -1, 0
    for l < n {
        for r+1 < n && nums[r+1] - nums[l] < k {
            r++
        }
        count += n - (r + 1) // считаем число пар, дающих в сумме >= k
        l++
    }
    return count
}

import java.util.*;

public class Solution {
    public Integer countPairs(List<Integer> nums, Integer k) {
        Collections.sort(nums);
        int n = nums.size();
        if (k == 0) {
            // формула количества пар i <= j: n * (n + 1) / 2
            return n * (n + 1) / 2;
        }
        int l = 0;
        int r = -1;
        int count = 0;
        while (l < n) {
            // Двигаем правый указатель, пока разница < k
            while (r + 1 < n && nums.get(r + 1) - nums.get(l) < k) {
                r += 1;
            }
            // считаем число пар, дающих в сумме >= k
            // с элементом nums[l] - это все элементы справа
            // от плавающего окна
            count += n - (r + 1);
            l += 1;
        }
        return count;
    }
}

from typing import List

def count_pairs(nums: List[int], k: int) -> int:
    nums.sort()
    n = len(nums)
    if k == 0:
        # формула количества пар i <= j: n * (n + 1) // 2
        return n * (n + 1) // 2
    l = 0
    r = -1
    count = 0
    while l < n:
        # Двигаем правый указатель, пока разница < k
        while r + 1 < n and nums[r + 1] - nums[l] < k:
            r += 1
        # считаем число пар, дающих в сумме >= k
        # с элементом nums[l] - это все элементы справа
        # от плавающего окна
        count += n - (r + 1)
        l += 1
    return count

/**
* @param {number[]} nums
* @param {number} k
* @returns {number}
*/
export function countPairs(nums, k) {
    nums.sort((a, b) => a - b);
    const n = nums.length;
    if (k === 0) {
        // формула количества пар i <= j: n * (n + 1) // 2
        return Math.floor(n * (n + 1) / 2);
    }
    let l = 0;
    let r = -1;
    let count = 0;
    while (l < n) {
        // Двигаем правый указатель, пока разница < k
        while (r + 1 < n && nums[r + 1] - nums[l] < k) {
            r += 1;
        }
        // считаем число пар, дающих в сумме >= k
        // с элементом nums[l] - это все элементы справа
        // от плавающего окна
        count += n - (r + 1);
        l += 1;
    }
    return count;
}