Codility: FrogRiverOne

FrogRiverOne

Find the earliest time when a frog can jump to the other side of a river.

Question

A small frog wants to get to the other side of a river. The frog is currently located at position 0, and wants to get to position X. Leaves fall from a tree onto the surface of the river.

You are given a non-empty zero-indexed array A consisting of N integers representing the falling leaves. A[K] represents the position where one leaf falls at time K, measured in minutes.

The goal is to find the earliest time when the frog can jump to the other side of the river. The frog can cross only when leaves appear at every position across the river from 1 to X.

For example, you are given integer X = 5 and array A such that:

  A[0] = 1
  A[1] = 3
  A[2] = 1
  A[3] = 4
  A[4] = 2
  A[5] = 3
  A[6] = 5
  A[7] = 4

In minute 6, a leaf falls into position 5. This is the earliest time when leaves appear in every position across the river.

Write a function:

int solution(int X, int A[], int N);

that, given a non-empty zero-indexed array A consisting of N integers and integer X, returns the earliest time when the frog can jump to the other side of the river.

If the frog is never able to jump to the other side of the river, the function should return −1.

For example, given X = 5 and array A such that:

  A[0] = 1
  A[1] = 3
  A[2] = 1
  A[3] = 4
  A[4] = 2
  A[5] = 3
  A[6] = 5
  A[7] = 4

the function should return 6, as explained above.

Assume that:

  • N and X are integers within the range [1..100,000];
  • each element of array A is an integer within the range [1..X].

Complexity:

  • expected worst-case time complexity is O(N);
  • expected worst-case space complexity is O(X), beyond input storage (not counting the storage required for input arguments).

Elements of input arrays can be modified.

 

Solution

My solution: https://codility.com/demo/results/demo2MK6RK-HDH/

//A: [1, 3, 1, 4, 2, 3, 5, 4]     X: 5       return:6
//A: [1, 3, 1, 3, 2, 1, 3]        X: 3       return:4
//A: [3]                          X: 5       return:-1
//A: [2, 2, 2, 2, 2]              X: 2       return:-1

#import <Foundation/NSDictionary.h> //Remember to import
-(int)solutionInt:(int)X WithArray:(NSArray *)A{
    //To record which leave has dropped on the position
    NSMutableDictionary *leavesDrop = [[NSMutableDictionary alloc] initWithCapacity:A.count];
    int remainingLeafDrop = X;
    for(int i= 0; i < A.count; i++){
        int index = [A[i] intValue]; //Position of leave dropped
        if(![leavesDrop objectForKey:[NSString stringWithFormat:@"%d", index]]){ //No leave drop at that position before
            [leavesDrop setValue:[NSNumber numberWithInt:1]
                       forKey:[NSString stringWithFormat:@"%d", index]];   //Add leave drop at that position
            remainingLeafDrop--;
        }
        
        if (remainingLeafDrop == 0) {
            return i;
        }
    }
    
    return -1;
}


Detected time complexity:
O(N)

 

Advertisements

Leave a Reply

Fill in your details below or click an icon to log in:

WordPress.com Logo

You are commenting using your WordPress.com account. Log Out / Change )

Twitter picture

You are commenting using your Twitter account. Log Out / Change )

Facebook photo

You are commenting using your Facebook account. Log Out / Change )

Google+ photo

You are commenting using your Google+ account. Log Out / Change )

Connecting to %s