Question

实施例。我有一个包含15个对象的数组。我想从给定的索引开始枚举。比如说从索引5开始，然后是上面的索引，索引在下面，上面，下面等等...我确实希望它能够回滚。

所以我的例子中的索引顺序是。 5,6,4,7,3,8,2,9,1,10,0,11,14,12,13

如果方法签名类似于以下行，那将会很棒，但我并不需要批准答案：

- (void)enumerateFromIndex:(NSUInteger)index wrapAroundAndGoBothWays:(void (^)(id obj, NSUInteger idx, BOOL *stop))block

如何做到这一点？想避免复制数组等。

在这篇文章中，我们使用没有环绕：Enumerate NSArray starting at givven index searching both ways (no wrap around)

Answer 1

借用@omz，这是包装变体，它更简单：

@implementation NSArray (Extensions)

- (void)enumerateFromIndex:(NSUInteger)index wrapAroundAndGoBothWays:(void (^)(id obj, NSUInteger idx, BOOL *stop))block
{
    BOOL stop = NO;
    NSUInteger actual = index;
    for (NSUInteger i = 0; i < self.count && !stop; i++) {
        actual += (2*(i%2)-1)*i;
        actual = (self.count + actual)%self.count;
        block([self objectAtIndex:actual], actual, &stop);
    }
}

@end

Answer 2

这是一个数学问题。有一个很好的解决方案。但是，它涉及提前对索引列表进行排序。

这个想法是将一个从0到15的整数放在一个圆上，然后按照它们在轴上出现的顺序取出这些元素。

由于在ObjC中这样做是如此繁琐，我提出了python解决方案：

from math import pi, cos

def circlesort(N, start):
    eps = 1e-8
    res = range(N)
    def f(x):
        return -cos(2*pi*(x-start-eps)/N)
    res.sort( lambda x,y:cmp(f(x), f(y)) )
    return res

然后

print circlesort(15, 5)

输出

[5, 6, 4, 7, 3, 8, 2, 9, 1, 10, 0, 11, 14, 12, 13]

这是期望的结果。

修改

好的，这是一个C实现：

#include <stdlib.h> #include <math.h> #define sign(x) ((x)>0?1:(x)<0?-1:0) void circlesort(int* values, int N, int start){ double f(int x) { return -cos(2*M_PI*((double)(x-start)-.25)/N); } int compare (const void * a, const void * b) { return sign( f(*(int*)a) - f(*(int*)b) ); } qsort (values, N, sizeof(int), compare); }

这将使长度为N的整数数组成圆圈。使用它如下：

int i, N = 15; int indexes[N]; for (i=0;i<N;i++) indexes[i] = i; circlesort(indexes, N, 5);

现在，数组indexes按所需顺序排序。因为有嵌套函数，所以应该将-fnested-functions添加到编译器标志。

编辑2

考虑到有一个更简单的解决方案（参见我的另一个答案），这个是相当学术性的。

枚举NSArray从给定索引开始搜索两种方式（环绕）

2 个答案: