From f05d888a0b621ca4e99e2b0fb6e23c097006fe41 Mon Sep 17 00:00:00 2001 From: Santo Cariotti Date: Tue, 28 May 2024 10:29:13 +0200 Subject: Init --- progs/a934.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) create mode 100644 progs/a934.py (limited to 'progs/a934.py') diff --git a/progs/a934.py b/progs/a934.py new file mode 100644 index 0000000..8dc5f52 --- /dev/null +++ b/progs/a934.py @@ -0,0 +1,12 @@ +def max_sum_pair_diff_lessthan_K(arr, N, K): + arr.sort() + dp = [0] * N + dp[0] = 0 + for i in range(1, N): + dp[i] = dp[i-1] + if (arr[i] - arr[i-1] < K): + if (i >= 2): + dp[i] = max(dp[i], dp[i-2] + arr[i] + arr[i-1]); + else: + dp[i] = max(dp[i], arr[i] + arr[i-1]); + return dp[N - 1] \ No newline at end of file -- cgit v1.2.3-18-g5258