Correctness and the Loop Invariant – HackerRank Solution

In this post, we will solve Correctness and the Loop Invariant HackerRank Solution. This problem (Correctness and the Loop Invariant) is a part of HackerRank Problem Solving series.

Objective

In the previous challenge, you wrote code to perform an Insertion Sort on an unsorted array. But how would you prove that the code is correct? I.e. how do you show that for any input your code will provide the right output?

Loop Invariant
In computer science, you could prove it formally with a loop invariant, where you state that a desired property is maintained in your loop. Such a proof is broken down into the following parts:

  • Initialization: It is true (in a limited sense) before the loop runs.
  • Maintenance: If it’s true before an iteration of a loop, it remains true before the next iteration.
  • Termination: It will terminate in a useful way once it is finished.

Insertion Sort’s Invariant

Say, you have some InsertionSort code, where the outer loop goes through the whole array A:

for(int i = 1; i < A.length; i++){
//insertion sort code

You could then state the following loop invariant:

At the start of every iteration of the outer loop (indexed with i), the subarray until ar[i] consists of the original elements that were there, but in sorted order.

To prove Insertion Sort is correct, you will then demonstrate it for the three stages:

  • Initialization – The subarray starts with the first element of the array, and it is (obviously) sorted to begin with.
  • Maintenance – Each iteration of the loop expands the subarray, but keeps the sorted property. An element V gets inserted into the array only when it is greater than the element to its left. Since the elements to its left have already been sorted, it means V is greater than all the elements to its left, so the array remains sorted. (In Insertion Sort 2 we saw this by printing the array each time an element was properly inserted.)
  • Termination – The code will terminate after i has reached the last element in the array, which means the sorted subarray has expanded to encompass the entire array. The array is now fully sorted.

You can often use a similar process to demonstrate the correctness of many algorithms. You can see these notes for more information.

Challenge

In the InsertionSort code below, there is an error. Can you fix it? Print the array only once, when it is fully sorted.

Input Format

There will be two lines of input:

  • s – the size of the array
  • arr – the list of numbers that makes up the array

Constraints

  • 1 <= s <= 1000
  • -1500 <= V <= 1500, Varr

Output Format

Output the numbers in order, space-separated on one line.

Sample Input

6
7 4 3 5 6 2

Sample Output

2 3 4 5 6 7

Explanation

The corrected code returns the sorted array.

Solution – Correctness and the Loop Invariant – HackerRank Solution

C++

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <assert.h>
#include <iostream>
using namespace std;
/* Head ends here */
void merge(int p[], int m, int q[], int n, int r[]){
    for(int rb=0, pf=0,qf=0;rb<m+n;rb++){
        if(pf<m&&qf<n){
            if(p[pf]<=q[qf]){
                r[rb]=p[pf];
                pf++;
            }
            else{
                r[rb]=q[qf];
                qf++;
            }
        }
        else if(pf<m){
            r[rb]=p[pf];
            pf++;
        }
        else{
            r[rb]=q[qf];
            qf++;
        }
    }
}

void mergesort(int*r,int n){
    if(n>1){
        int p[n/2],q[n-n/2];
        for(int i=0;i<n/2;i++)p[i]=r[i];
        for(int i=n/2;i<n;i++)q[i-n/2]=r[i];
        mergesort(p,n/2);
        mergesort(q,n-n/2);
        merge(p,n/2,q,n-n/2,r);
    }
}
/* Tail starts here */
int main(void) {
   
   int _ar_size;
cin>>_ar_size;
int _ar[_ar_size], _ar_i;
for(_ar_i = 0; _ar_i < _ar_size; _ar_i++) { 
   cin>>_ar[_ar_i];
}

mergesort(_ar,_ar_size);
   
    for(int i=0;i<_ar_size;i++)cout<<_ar[i]<<" ";
    cout<<endl;
   return 0;
}

Python

def insertion_sort(l):
    for i in range(1, len(l)):
        j = i-1
        key = l[i]
        while (j >= 0) and (l[j] > key):
            l[j+1] = l[j]
            j -= 1
        l[j+1] = key


m = int(input().strip())
ar = [int(i) for i in input().strip().split()]
insertion_sort(ar)
print(" ".join(map(str,ar)))

Java

import java.io.*;
import java.util.*;

public class Solution {

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int numElements = in.nextInt();
        int[] array = new int[numElements];
        for (int i = 0; i < array.length; i++) {
            array[i] = in.nextInt();
        }

        sort(array);
    }

    public static void sort(int[] array) {
        for (int i = 1; i < array.length; i++) {
            int index = i;
            while (index > 0 && array[index] < array[index - 1]) {
                int temp = array[index];
                array[index] = array[index - 1];
                array[index - 1] = temp;
                index--;
            }
        }
        
        printArray(array);
    }

    public static void printArray(int[] array) {
        for (int i = 0; i < array.length; i++) {
            System.out.print(array[i] + " ");
        }
        System.out.println();
    }

}

Note: This problem (Correctness and the Loop Invariant) is generated by HackerRank but the solution is provided by CodingBroz. This tutorial is only for Educational and Learning purpose.

Leave a Comment

Your email address will not be published. Required fields are marked *