Objective-C 关联引用的实例（59）

最新推荐文章于 2024-06-10 00:15:00 发布

stack-0

最新推荐文章于 2024-06-10 00:15:00 发布

阅读量295

点赞数

CC 4.0 BY-SA版权

分类专栏： Objective-C 读书笔记

本文链接：https://blog.youkuaiyun.com/qq_33033983/article/details/61616487

Objective-C 同时被 2 个专栏收录

31 篇文章

订阅专栏

读书笔记

24 篇文章

订阅专栏

本文介绍了一个Objective-C的NSArray类别扩展方法，通过该方法可以实现从NSArray中随机选取一个元素的功能。该方法利用了随机数生成器，并考虑到了数组为空或只有一个元素的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#import <Foundation/NSArray.h>

@interface NSArray (Random)

- (id)anyOne;

@end

#import "NSArray+Random.h"
#import <objc/runtime.h>

@implementation NSArray (Random)

static char prevKey; //定义键使用的地址变了

static int random_value(void) {
    static unsigned long rnd = 201008;
    rnd = rnd * 1103515245UL + 12345;
    return (int)((rnd >> 16) & 0x7fff);
}

- (id)anyOne
{
    id item;
    NSUInteger count = [self count];
    if (count == 0)
        return nil;
    if (count == 1)
        item = [self lastObject];
    else {
        id prev = objc_getAssociatedObject(self, &prevKey);
        NSUInteger index = random_value() % count;
        item = [self objectAtIndex:index];
        if (item == prev) {
            if (++index >= count)
                index = 0;
            item = [self objectAtIndex:index];
        }
    }
    objc_setAssociatedObject(self, &prevKey, item, OBJC_ASSOCIATION_RETAIN);
    return item;
}

@end

#import <Foundation/Foundation.h>
#import "NSArray+Random.h"

int main(void)
{
    @autoreleasepool {
        id arr1 = [NSArray arrayWithObjects:@"A", @"B", @"C", @"D", @"E", @"F", @"G", @"H", nil];
        id arr2 = [NSMutableArray arrayWithObjects:@"1", @"2", @"3", @"4", nil];
        for (int i = 5; i < 20; i++) {
            @autoreleasepool {
                printf("%s %s\n",[[arr1 anyOne] UTF8String], [[arr2 anyOne] UTF8String]);
            }
        }
    }
    return 0;
}