Given an array a[] consisting of N positive integers, and an integer K, the task is to find the minimum number of operations required to make the sum of adjacent elements less than or equal to K, where, one operation involves decreasing any array element by 1. For every ith element in the given array, the operations can be performed 0 to a[i] times. Since the answer can be large, compute it modulo 109 + 7.
Examples:
Input: a[] = {11, 3, 13, 10, 8, 17, 22}, K = 14
Output: 34
Explanation:
Minimum number of operations required to obtain the desired arrangement is as follows:
- Reduce a[2] by 2
- Reduce a[3] by 7
- Reduce a[5] by 11
- Reduce a[6] by 14
The given array is modified to the following arrangement = {11, 3, 11, 3, 8, 6, 8}
Total number of operations is 2 + 5 + 7 + 11 + 18 = 34.
Input: a[] = {1000000000, 1000000000, 1000000000, 1000000000}, K = 0
Output3: 999999979
Explanation:
Since the sum of adjacent pairs is required to be 0, all elements in the array need to be reduced to 0.
Therefore, the answer is sum of array % (109 + 7).
Sum of array is 4000000000
Therefore, the required answer is 4000000000 % 109 + 7 = 999999979
Approach:
Follow the steps below to solve the problem:
- Iterate over the array a[] and for each adjacent pair, check if their sum is less than or equal to K. If found to be true, no changes are required.
- For pairs with sum greater than K, follow the steps below:
- If the first element of pair exceeds K, make the value of the first element in the pair equal to K. Increase the number of operations required by value of first element – K and update the value of the first element of the pair to K.
- Now, apply the sum of pair – K operations on the second element to ensure that the sum of pair is equal to K, and update the second element of pair to K – value of first element.
- Repeat the above steps for all the elements and print the number of operations calculated.
Below is the implementation of the above approach:
C++
#include<bits/stdc++.h>
using namespace std;
int minimum_required_operations( int arr[],
int n, int k)
{
int answer = 0;
long long mod = 1000000007;
for ( int i = 0; i < n - 1; i++)
{
if (arr[i] + arr[i + 1] > k)
{
if (arr[i] > k)
{
answer += (arr[i] - k);
arr[i] = k;
}
answer += (arr[i] + arr[i + 1]) - k;
arr[i + 1] = (k - arr[i]);
answer %= mod;
}
}
return answer;
}
int main()
{
int a[] = { 9, 50, 4, 14, 42, 89 };
int k = 10;
int n = sizeof (a) / sizeof (a[0]);
cout << (minimum_required_operations(a, n, k));
return 0;
}
|
Java
import java.util.*;
class GFG{
static int minimum_required_operations( int arr[],
int n, int k)
{
int answer = 0 ;
long mod = 1000000007 ;
for ( int i = 0 ; i < n - 1 ; i++)
{
if (arr[i] + arr[i + 1 ] > k)
{
if (arr[i] > k)
{
answer += (arr[i] - k);
arr[i] = k;
}
answer += (arr[i] + arr[i + 1 ]) - k;
arr[i + 1 ] = (k - arr[i]);
answer %= mod;
}
}
return answer;
}
public static void main(String[] args)
{
int a[] = { 9 , 50 , 4 , 14 , 42 , 89 };
int k = 10 ;
int n = a.length;
System.out.print(
minimum_required_operations(a, n, k));
}
}
|
Python3
def minimum_required_operations(arr, n, k):
answer = 0
mod = 10 * * 9 + 7
for i in range (n - 1 ):
if arr[i] + arr[i + 1 ] > k:
if arr[i] > k:
answer + = (arr[i] - k)
arr[i] = k
answer + = (arr[i] + arr[i + 1 ]) - k
arr[i + 1 ] = (k - arr[i])
answer % = mod
return answer
a = [ 9 , 50 , 4 , 14 , 42 , 89 ]
k = 10
print (minimum_required_operations(a, len (a), k))
|
C#
using System;
class GFG{
static int minimum_required_operations( int [] arr, int n,
int k)
{
int answer = 0;
long mod = 1000000007;
for ( int i = 0; i < n - 1; i++)
{
if (arr[i] + arr[i + 1] > k)
{
if (arr[i] > k)
{
answer += (arr[i] - k);
arr[i] = k;
}
answer += (arr[i] + arr[i + 1]) - k;
arr[i + 1] = (k - arr[i]);
answer = ( int )(answer % mod);
}
}
return answer;
}
public static void Main(String[] args)
{
int [] a = { 9, 50, 4, 14, 42, 89 };
int k = 10;
int n = a.Length;
Console.Write(minimum_required_operations(a, n, k));
}
}
|
Javascript
<script>
function minimum_required_operations(arr,n,k)
{
let answer = 0;
let mod = 1000000007;
for (let i = 0; i < n - 1; i++)
{
if (arr[i] + arr[i + 1] > k)
{
if (arr[i] > k)
{
answer += (arr[i] - k);
arr[i] = k;
}
answer += (arr[i] + arr[i + 1]) - k;
arr[i + 1] = (k - arr[i]);
answer %= mod;
}
}
return answer;
}
let a = [ 9, 50, 4, 14, 42, 89 ];
let k = 10;
let n = a.length;
document.write(
minimum_required_operations(a, n, k));
</script>
|
Time Complexity: O(N)
Auxiliary Space: O(1)