自己动手写MutableArray

自定义的可变数组

学过数据结构的人应该了解，数据结构是用来在内存中存储数据用的，在大学期间学习线性表，链表，十字链表，图，二叉树等的时候，都是要自己写创建，遍历，插入，删除等函数的实现。然而后面学了面向对象语言后，系统都自己封装好了，可以直接使用，条用方法即可。为了强化自己的动手编程能力，决定自己手写一个object-c中的可变数组，用链表实现。

申明部分

#import <Foundation/Foundation.h>

@interface HTCMutableArray : NSObject

//  计算数组中对象的个数
@property (nonatomic,assign,readonly) NSInteger count;

//  类方法创建对象
+ (instancetype)array;

//  添加对象到数组中
- (void)addObject:(id)object;

//  根据索引得到对象
- (id)objectAtIndex:(NSInteger)index;

//  插入对象到索引为index的数组中
- (void)insertObject:(id)object atIndex:(NSInteger)index;

//  从数组中移除索引为index的对象
- (void)removeObjectAtIndex:(NSInteger)index;

//  遍历数组中的对象，输出其地址
- (void)ergodicArray;

@end

实现部分

#import "HTCMutableArray.h"


/**
 *  自定义可变数组，使用数据结构中的链表结构完成
 */
@interface HTCMutableArray ()

@property (nonatomic,strong) id object;

@property (nonatomic,strong) HTCMutableArray *next;

@end

@implementation HTCMutableArray

- (instancetype)init{
//    1.创建array对象时，记为头结点，不存储数据，
    self = [super init];
    _count = 0;
    self.object = nil;
    self.next = nil;
    return self;
}

//  类方法创建对象
+ (instancetype)array{
    HTCMutableArray *array = [[HTCMutableArray alloc] init];

    return array;
}

//  添加对象到数组中
- (void)addObject:(id)object{
    HTCMutableArray *array = [HTCMutableArray array];
    HTCMutableArray *temp = self;
//    1.遍历链表，如果当前节点的next指针为nil，结束循环
    while(temp.next){
        temp = temp.next;
    }

//    2.将尾节点的next指针赋给要添加的对象
    temp.next = array;
    array.object = object;
//    3.将要添加的对象的next指针设为nil
    array.next = nil;

    _count++;

}

//  根据索引得到对象
- (id)objectAtIndex:(NSInteger)index{
    NSInteger tempIndex = 0;
//    1.取出头结点next（即第一个节点，因为头结点没有用来存储数据，只是一个起始标识）
    HTCMutableArray *temp = self.next;
//    2.遍历链表，
    while(temp){

        if (index == tempIndex) break;
        tempIndex++;
        temp = temp.next;
    }
//    3.返回节点中的object对象
    return temp.object;
}

//  插入对象到索引为index的数组中
- (void)insertObject:(id)object atIndex:(NSInteger)index{

    if (index > _count) {
        return;
    }
    NSInteger tempIndex = 0;
    HTCMutableArray *temp = self.next;
    while (temp) {
        if (index == tempIndex) {

            HTCMutableArray *array = [HTCMutableArray array];
            array.object = object;
            array.next = temp.next;
            temp.next = array;
            _count++;
            break;
        }
        temp = temp.next;
        tempIndex++;
    }
}

//  从数组中移除索引为index的对象
- (void)removeObjectAtIndex:(NSInteger)index{
    HTCMutableArray *temp = self.next;
    HTCMutableArray *remove = nil;
    NSInteger tempIndex = 0;
    while (temp) {
        if (tempIndex == (index - 1)) {

            remove = temp.next;
            temp.next = temp.next.next;
            remove = nil;
            _count--;
            break;
        }
        tempIndex++;
        temp = temp.next;
    }
}

//  遍历数组中的对象，输出其地址
- (void)ergodicArray{
    HTCMutableArray *temp = self.next;
    while (temp) {

        NSLog(@"object pointer = %p",temp.object);

        temp = temp.next;
    }
}

@end

使用案例

#import <Foundation/Foundation.h>
#import "HTCMutableArray.h"

int main(int argc, const char * argv[]) {
    @autoreleasepool {
        // insert code here...

//        1.定义3个对象
        NSObject *obj1 = [[NSObject alloc] init];
        NSObject *obj2 = [[NSObject alloc] init];
        NSObject *obj3 = [[NSObject alloc] init];

//        2.创建自定义的数组对象
        HTCMutableArray *array = [HTCMutableArray array];

//        3.将对象添加到数组中
        [array addObject:obj1];
        [array addObject:obj2];
        [array addObject:obj3];

//        4.打印数组中对象的个数
        NSLog(@"count = %ld",array.count);

//        5.插入一个新的对象
        NSObject *obj5 = [[NSObject alloc] init];
        [array insertObject:obj5 atIndex:1];

//        6.根据索引值获得对象
        NSObject *obj4 = [array objectAtIndex:1];

//        7.输出5个对象的地址
        NSLog(@"obj1=%p,obj2=%p,obj3=%p,obj4=%p,obj5=%p",obj1,obj2,obj3,obj4,obj5);

//        8.遍历数组，输出其地址，
        [array ergodicArray];
    }
    return 0;
}

运行情况

第一次写blog，后面会继续坚持下去的，加油！