Az Flights Optimization, Objective-C

Language: Layout:

//gcc 5.0.4

#import <Foundation/Foundation.h>
#include <time.h>
#include <stdlib.h>

void swap (NSInteger *indexes, NSInteger a, NSInteger b) {
    NSInteger temp = indexes [a];
    indexes [a] = indexes [b];
    indexes [b] = temp;
}

NSInteger partition (NSArray *flights, NSInteger count, NSInteger *indexes, NSInteger low, NSInteger high) {
    srand (time(NULL)); 
    NSInteger random = low + rand() % (high - low); 
    swap (indexes, random, high);
    NSInteger pivot = ((NSNumber*)[((NSArray*)[flights objectAtIndex: indexes [high]]) objectAtIndex: 1]).integerValue;
    NSInteger left = low, right = high - 1;
    
    while (left < right) {
        while (left < count && ((NSNumber*)[((NSArray*)[flights objectAtIndex: indexes [left]]) objectAtIndex: 1]).integerValue < pivot)
            left++;
        
         while (right >= 0 && ((NSNumber*)[((NSArray*)[flights objectAtIndex: indexes [right]]) objectAtIndex: 1]).integerValue >= pivot)
            right--;
        
        if (left < right) {
            swap (indexes, left, right);
            left++;
            right--;
        }
    }
    
    swap (indexes, left, high);
    
    return left;
}

void quickSort (NSArray *flights, NSInteger count, NSInteger *indexes, NSInteger low, NSInteger high) {
    if (low < high) {
           NSInteger pivot = partition (flights, count, indexes, low, high);
           quickSort (flights, count, indexes, low, pivot - 1);
        quickSort (flights, count, indexes, pivot + 1, high);
    }
    
    
    
}

NSInteger *sortFlights (NSArray *flights) {
    NSInteger count = flights.count, *result = (NSInteger *) malloc (sizeof (NSInteger) * count), index;
    
    for (index = 0; index < count; index++)
        result [index] = index;
    
    quickSort (flights, count, result, 0, count - 1);
    
    return result;
}

NSArray* optimizeFlightDictionary (NSArray *forwardFlights, NSArray *returnFlights, NSInteger maxDistance) {
    NSMutableArray *result = [NSMutableArray array];
    
    NSInteger forwardFlightsCount = forwardFlights.count, index;
    NSInteger *forwardSortedFlights = sortFlights (forwardFlights);
    
    NSLog (@"Forward");
    for (index = 0; index < forwardFlightsCount; index++)
        NSLog (@"index: %ld fw: %ld value: %ld", index, forwardSortedFlights [index], ((NSNumber*)[((NSArray*)[forwardFlights objectAtIndex: forwardSortedFlights [index]]) objectAtIndex: 1]).integerValue);
    
    NSInteger returnFlightsCount = returnFlights.count;
    NSInteger *returnSortedFlights = sortFlights (returnFlights);
    
    NSLog (@"Return");
    for (index = 0; index < returnFlightsCount; index++)
        NSLog (@"index: %ld fw: %ld value: %ld", index, returnSortedFlights [index], ((NSNumber*)[((NSArray*)[returnFlights objectAtIndex: returnSortedFlights [index]]) objectAtIndex: 1]).integerValue);
    
    
    return result;
}

int main (int argc, const char * argv[])
{
    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
    
    NSMutableArray *forwardFlights = [NSMutableArray array], *returnFlights = [NSMutableArray array], *workingElement;
    
    
    workingElement = [NSMutableArray array];
    [workingElement addObject: [NSNumber numberWithInteger: 1]];  //Doesn't work @1
    [workingElement addObject: [NSNumber numberWithInteger: 1000]];  //Doesn't work @1000
    [forwardFlights addObject: workingElement];
    
    workingElement = [NSMutableArray array];
    [workingElement addObject: [NSNumber numberWithInteger: 2]];
    [workingElement addObject: [NSNumber numberWithInteger: 1500]];
    [forwardFlights addObject: workingElement];
    
    workingElement = [NSMutableArray array];
    [workingElement addObject: [NSNumber numberWithInteger: 3]];
    [workingElement addObject: [NSNumber numberWithInteger: 500]];
    [forwardFlights addObject: workingElement];
    
    workingElement = [NSMutableArray array];
    [workingElement addObject: [NSNumber numberWithInteger: 4]];
    [workingElement addObject: [NSNumber numberWithInteger: 1200]];
    [forwardFlights addObject: workingElement];
    
    workingElement = [NSMutableArray array];
    [workingElement addObject: [NSNumber numberWithInteger: 5]];
    [workingElement addObject: [NSNumber numberWithInteger: 2500]];
    [forwardFlights addObject: workingElement];
    
    workingElement = [NSMutableArray array];
    [workingElement addObject: [NSNumber numberWithInteger: 1]];
    [workingElement addObject: [NSNumber numberWithInteger: 1200]];
    [returnFlights addObject: workingElement];
    
    workingElement = [NSMutableArray array];
    [workingElement addObject: [NSNumber numberWithInteger: 2]];
    [workingElement addObject: [NSNumber numberWithInteger: 700]];
    [returnFlights addObject: workingElement];
    
    workingElement = [NSMutableArray array];
    [workingElement addObject: [NSNumber numberWithInteger: 3]];
    [workingElement addObject: [NSNumber numberWithInteger: 900]];
    [returnFlights addObject: workingElement];
    
    workingElement = [NSMutableArray array];
    [workingElement addObject: [NSNumber numberWithInteger: 4]];
    [workingElement addObject: [NSNumber numberWithInteger: 500]];
    [returnFlights addObject: workingElement];
    
    workingElement = [NSMutableArray array];
    [workingElement addObject: [NSNumber numberWithInteger: 5]];
    [workingElement addObject: [NSNumber numberWithInteger: 1900]];
    [returnFlights addObject: workingElement];
    
     workingElement = [NSMutableArray array];
    [workingElement addObject: [NSNumber numberWithInteger: 6]];
    [workingElement addObject: [NSNumber numberWithInteger: 700]];
    [returnFlights addObject: workingElement];
    
     workingElement = [NSMutableArray array];
    [workingElement addObject: [NSNumber numberWithInteger: 7]];
    [workingElement addObject: [NSNumber numberWithInteger: 1700]];
    [returnFlights addObject: workingElement];
   
    NSInteger maxDistance = 3000;
    
    NSLog (@"forwardFlights: %@",forwardFlights);
    NSLog (@"returnFlights: %@",returnFlights);
    NSLog (@"maxDistance: %ld",maxDistance);
    
    optimizeFlightDictionary (forwardFlights, returnFlights, maxDistance);
    
    
    [pool drain];
    return 0;
}

Show compiler warnings [ + ] Compiler args [ + ] Show input
Compilation time: 0.32 sec, absolute running time: 0.09 sec, cpu time: 0.04 sec, memory peak: 9 Mb, absolute service time: 0,42 sec