iOS 中类似 SKU 刷选，利用质数实现

在我们添加购物车的时候，有时某些商品的属性需要进行刷选，而且类似 Color 和 Size 之间还有多种选择，此时会有可选和可用的状态，类似如下。

类似淘宝添加时的属性

经过小伙伴的研究和提醒，觉的我们可以利用两个质数之间的积是唯一的，来实现。

三种状态：选中，不选中，不可用。
也就是当前同一行有选中和普通的状态；其他行则是可用和不可用的状态。
注意看假数据构成，这样就很容易一目了然的

假数据

具体初次的实现效果

Demo 演示效果

PS：下面先造假数据，直接先实现，现实中要复杂很多。

#import "ViewController.h"

static NSString *const kPQTestCollectionCellIden = @"TestCollectionCellIden";
static const NSInteger kPQTestButtonTag = 100;

@interface ViewController ()<UICollectionViewDelegate, UICollectionViewDataSource, UICollectionViewDelegateFlowLayout>

@property (weak, nonatomic) IBOutlet UICollectionView *collectionView;
/**
 总的数据模型，来自后台
 */
@property (nonatomic, strong) NSArray<NSArray *> *dataArray;
/**
 可用的组合积的数组，来自后台
 */
@property (nonatomic, strong) NSArray *userAvailabilityArray;
/**
 选中的积，临时变量
 */
@property (nonatomic, strong) NSMutableArray *chooseSelectArray;

@end

@implementation ViewController

- (void)viewDidLoad {
    [super viewDidLoad];
    
    self.dataArray = @[
                       @[@"2",@"3", @"5"],
                       @[@"7",@"11",@"13",@"17"],
                       @[@"19",@"23",@"29"],
                       @[@"37",@"41", @"43",@"47"]
                       ];
    
    
    /**
     
     假设这些选中是 OK 的, 保证每一个都有的选
     2 * 7 * 19 * 37 = 9842
     2 * 11 * 23 * 37 = 18722
     2 * 17 * 19 * 43 = 27778
     
     3 * 13 * 23 * 43 = 38571
     3 * 17 * 29 * 43 = 63597
     3 * 17 * 29 * 47 = 69513
     3 * 7 * 23 * 47 = 22701
     
     5 * 11 * 19 * 37 = 38665
     5 * 17 * 23 * 43 = 84065
     5 * 7 * 23 * 41 = 33005
     5 * 17 * 29 * 47 = 115855
     
     
     */
    self.userAvailabilityArray = @[@"9842",@"18722",@"27778",@"38571",@"63597",@"69513",@"22701",@"38665",@"33005",@"84065",@"115855"];
    
    [self.collectionView reloadData];
    
}

- (void)changeButtonStatus:(UIButton *)chooseButton {
    // 对选中状态取反
    chooseButton.selected = !chooseButton.selected;
    // 获取 button 的具体信息
    NSInteger selectSection = chooseButton.tag / kPQTestButtonTag  - 1;
    NSInteger selectRow = chooseButton.tag % kPQTestButtonTag - 1;
    
    if (chooseButton.selected) {
        // 选中的状态
        [self selectSomeButtonWithSection:selectSection row:selectRow chooseButtonTag:chooseButton.tag];
    }
    else {
        // 取反的状态
        [self unSelectSomeButtonWithSection:selectSection row:selectRow chooseButtonTag:chooseButton.tag];
    }
}

/**
 选中某个
 */
- (void)selectSomeButtonWithSection:(NSInteger)selectSection row:(NSInteger)selectRow chooseButtonTag:(NSInteger)chooseButtonTag {
    // 先判断如果是同一行的话
    [self.dataArray enumerateObjectsUsingBlock:^(NSArray * _Nonnull obj, NSUInteger sectionIdx, BOOL * _Nonnull sectionStop) {
        if (selectSection == sectionIdx) {
            // 对同一种类型进行互斥
            [self.dataArray[selectSection] enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger rowIdx, BOOL * _Nonnull stop) {
                // 获取与之对应的 Cell 中的 Button
                UIButton *button = [self getCellWithSection:selectSection row:rowIdx];
                // 假如是同一行的其他标签，而且是选中状态
                if (button.tag != chooseButtonTag && button.selected) {
                    // 将上一个选中设置为普通
                    button.selected = NO;
                    // 同时移除上一次的选中的值
                    [self.chooseSelectArray removeObject:self.dataArray[selectSection][rowIdx]];
                }
            }];
            *sectionStop = YES;
        }
    }];
    // 再判断其他种行种类的, 是否 可用
    [self.chooseSelectArray addObject:self.dataArray[selectSection][selectRow]];
    NSArray *useArray = [self getCanUseArrayWithSelect:YES];
    [self.dataArray enumerateObjectsUsingBlock:^(NSArray * _Nonnull obj, NSUInteger sectionIdx, BOOL * _Nonnull stop) {
        if (selectSection != sectionIdx) {
            // 不是选中的这一行，则刷新 是否可用
            [self.dataArray[sectionIdx] enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger rowIdx, BOOL * _Nonnull rowStop) {
                [useArray enumerateObjectsUsingBlock:^(NSString *useStr, NSUInteger useIdx, BOOL * _Nonnull useStop) {
                    // 获取与之对应的 Cell 中的 Button
                    UIButton *button = [self getCellWithSection:sectionIdx row:rowIdx];
                    // 是否可用
                    button.enabled = ([useStr integerValue] % [obj integerValue] == 0);
                    if (button.enabled) {
                        * useStop = YES;
                    }
                }];
            }];
        }
    }];
}

/**
 取反某个
 */
- (void)unSelectSomeButtonWithSection:(NSInteger)selectSection row:(NSInteger)selectRow chooseButtonTag:(NSInteger)chooseButtonTag {
    [self.chooseSelectArray removeObject:self.dataArray[selectSection][selectRow]];
    NSArray *useArray = [self getCanUseArrayWithSelect:NO];
    [self.dataArray enumerateObjectsUsingBlock:^(NSArray * _Nonnull obj, NSUInteger sectionIdx, BOOL * _Nonnull stop) {
        if (selectSection != sectionIdx) {
            [self.dataArray[sectionIdx] enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger rowIdx, BOOL * _Nonnull rowStop) {
                [useArray enumerateObjectsUsingBlock:^(NSString *useStr, NSUInteger useIdx, BOOL * _Nonnull useStop) {
                    // 获取与之对应的 Cell 中的 Button
                    UIButton *button = [self getCellWithSection:sectionIdx row:rowIdx];
                    // 是否可用
                    button.enabled = ([useStr integerValue] % [obj integerValue] == 0);
                    if (button.enabled) {
                        * useStop = YES;
                    }
                }];
            }];
        }
        else {
            [self.dataArray[selectSection] enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger rowIdx, BOOL * _Nonnull rowStop) {
                // 获取与之对应的 Cell 中的 Button
                UIButton *button = [self getCellWithSection:selectSection row:rowIdx];
                // 假如是其他标签，而且是选中状态
                if (button.tag == chooseButtonTag && !button.selected) {
                    // 同时移除上一次的选中的值
                    [self.chooseSelectArray removeObject:self.dataArray[selectSection][rowIdx]];
                    *rowStop = YES;
                }
            }];
        }
    }];
    
}

/**
 获取相对应的 Cell 中的 Button 按钮
 */
- (UIButton *)getCellWithSection:(NSInteger)section row:(NSInteger)row {
    // 获取与之对应的 Cell
    UICollectionViewCell *cell = [self.collectionView cellForItemAtIndexPath:[NSIndexPath indexPathForRow:row inSection:section]];
    // 获取与之对应的 Button
    UIButton *button = (UIButton *)[cell viewWithTag:(kPQTestButtonTag * (section + 1) + (row + 1))];
    return button;
}


/**
 获取已经选中的 Value 乘积
 */
- (long long)getChooseSelectValue {
    __block long long hadChooseValue = 1;
    [self.chooseSelectArray enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
        hadChooseValue *= [obj integerValue];
    }];
    return hadChooseValue;
}

/**
 获取可用的 数组 Value
 */
- (NSArray *)getCanUseArrayWithSelect:(BOOL)select {
    long long chooseAllValue = [self getChooseSelectValue];
    /**
     chooseAllValue == 1 表示为空
     self.chooseSelectArray.count == 1 && !select： 取反时只剩下一个的时候
     */
    if (chooseAllValue == 1 || (self.chooseSelectArray.count == 1 && !select)) {
        return self.userAvailabilityArray;
    }
    NSMutableArray *useArray = [NSMutableArray array];
    [self.userAvailabilityArray enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
        NSInteger useInteger = [obj integerValue];
        if ((useInteger % chooseAllValue) == 0) {
            [useArray addObject:obj];
        }
    }];
    return useArray.copy;
}

#pragma mark - Delegate
- (NSInteger)numberOfSectionsInCollectionView:(UICollectionView *)collectionView {
    return  self.dataArray.count;
}

- (NSInteger)collectionView:(UICollectionView *)collectionView numberOfItemsInSection:(NSInteger)section {
    return self.dataArray[section].count;;
}

- (UICollectionViewCell *)collectionView:(UICollectionView *)collectionView cellForItemAtIndexPath:(NSIndexPath *)indexPath {
    UICollectionViewCell *cell = [collectionView dequeueReusableCellWithReuseIdentifier:kPQTestCollectionCellIden forIndexPath:indexPath];
    // 显示 �UIButton
    UIButton *button = [UIButton buttonWithType:UIButtonTypeCustom];
    button.frame = cell.bounds; // 此处先写死，size 先设置个 60、50
    [button setTitle:(self.dataArray[indexPath.section][indexPath.row]) forState:UIControlStateNormal];
    [button setTitleColor:[UIColor whiteColor] forState:UIControlStateNormal];
    [button pq_setBackgroundColor:[UIColor orangeColor] state:UIControlStateSelected];
    [button pq_setBackgroundColor:[UIColor blackColor] state:UIControlStateNormal];
    [button pq_setBackgroundColor:[UIColor grayColor] state:UIControlStateDisabled];
    button.tag = kPQTestButtonTag * (indexPath.section + 1) + (indexPath.row + 1);
    [button addTarget:self action:@selector(changeButtonStatus:) forControlEvents:UIControlEventTouchUpInside];
    [cell addSubview:button];
    
    return cell;
}

#pragma mark - Getter
- (NSMutableArray *)chooseSelectArray {
    if (!_chooseSelectArray) {
        _chooseSelectArray = [NSMutableArray array];
    }
    return _chooseSelectArray;
}

@end

@implementation UIButton (PQBackgroundColor)

- (void)pq_setBackgroundColor:(UIColor *)color state:(UIControlState)state {
    [self setBackgroundImage:[self pq_imageWithColor:color] forState:state];
}

- (UIImage *)pq_imageWithColor:(UIColor *)color {
    CGRect rect = CGRectMake(0.0f, 0.0f, 1.0f, 1.0f);
    UIGraphicsBeginImageContext(rect.size);
    CGContextRef context = UIGraphicsGetCurrentContext();
    CGContextSetFillColorWithColor(context, [color CGColor]);
    CGContextFillRect(context, rect);
    
    UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    
    return image;
}

@end

以上 Demo 是直接用质数代替显示的，真实中需要通过 Model 来配合，另外以下几点需要注意下：

核心是任何两个质数的积是唯一的！！！
现实中注意和后台数据的协商，Model 组合怎样最方便，这个可以好好商量。
注意反选时，一定要多加测试，这里很容易留下坑。

当然上面 Demo 中的遍历很多，还可以很多优化，算法此处可优化的空间很大，欢迎指导。

由于上述仍然有问题，不够严谨，在不优先考虑算法的前提下，继续优化。

之前的思路是：

同行，做置反效果，就是选中和不选中的排序（select 和 normal的区别）
其他行，拿到选中的乘积看是否被整除否（enable 和 able 的区别）

现在的思路是：

正选：添加

先同一行：置反，这个需要单独提出来。
其他行：先判断此行有没有被选中，默认被选中的值为 1（hadSelectValue），然后已经选中的乘积（allValue），当前选中的值的 (nowItemValue)。
（allValue / hadSelectValue * nowItemValue）看是否可以被整除？

反选：移除

同一行：置反
其他行：同正选一样。

改进一、逻辑优化

效果二

/**
 选中某个
 */
- (void)selectSomeButtonWithSection:(NSInteger)selectSection row:(NSInteger)selectRow chooseButtonTag:(NSInteger)chooseButtonTag {
    // 先判断如果是同一行的话
    [self.dataArray enumerateObjectsUsingBlock:^(NSArray * _Nonnull obj, NSUInteger sectionIdx, BOOL * _Nonnull sectionStop) {
        if (selectSection == sectionIdx) {
            // 对同一种类型进行互斥
            [self.dataArray[selectSection] enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger rowIdx, BOOL * _Nonnull stop) {
                // 获取与之对应的 Cell 中的 Button
                UIButton *button = [self getCellWithSection:selectSection row:rowIdx];
                // 假如是同一行的其他标签，而且是选中状态
                if (button.tag != chooseButtonTag && button.selected) {
                    // 将上一个选中设置为普通
                    button.selected = NO;
                    // 同时移除上一次的选中的值
                    [self.chooseSelectArray removeObject:self.dataArray[selectSection][rowIdx]];
                }
            }];
            *sectionStop = YES;
        }
    }];
    // 再判断其他种行种类的, 是否 可用
    [self.chooseSelectArray addObject:self.dataArray[selectSection][selectRow]];
    NSInteger allValueIntger = [self getChooseSelectValue];
    [self.dataArray enumerateObjectsUsingBlock:^(NSArray * _Nonnull obj, NSUInteger sectionIdx, BOOL * _Nonnull stop) {
        if (selectSection != sectionIdx) {
            // 不是选中的这一行，则刷新 是否可用
            __block NSInteger hadSelectValue = 1;
            if (self.chooseSelectArray.count > 1) {
                [self.dataArray[sectionIdx] enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger rowIdx, BOOL * _Nonnull rowStop) {
                    if ([self.chooseSelectArray containsObject:obj]) {
                        hadSelectValue = [obj integerValue];
                        *rowStop = YES;
                    }
                }];
            }
            [self.dataArray[sectionIdx] enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger rowIdx, BOOL * _Nonnull rowStop) {
                // 获取与之对应的 Cell 中的 Button
                UIButton *button = [self getCellWithSection:sectionIdx row:rowIdx];
                // 获取当前的 Value
                NSInteger buttonValueInter = [self.dataArray[sectionIdx][rowIdx] integerValue];
                // 就是没有选中
                if (!button.selected) {
                    [self.userAvailabilityArray enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
                        button.enabled = NO;
                        if (([obj integerValue] % (allValueIntger * buttonValueInter / hadSelectValue)) == 0) {
                            button.enabled = YES;
                            *stop = YES;
                        }
                    }];
                }
            }];
        }
    }];
}

/**
 取反某个
 */
- (void)unSelectSomeButtonWithSection:(NSInteger)selectSection row:(NSInteger)selectRow chooseButtonTag:(NSInteger)chooseButtonTag {
    [self.chooseSelectArray removeObject:self.dataArray[selectSection][selectRow]];
    NSInteger allValueIntger = [self getChooseSelectValue];
    [self.dataArray enumerateObjectsUsingBlock:^(NSArray * _Nonnull obj, NSUInteger sectionIdx, BOOL * _Nonnull stop) {
        if (selectSection != sectionIdx) {
            [self.dataArray[sectionIdx] enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger rowIdx, BOOL * _Nonnull rowStop) {
                __block NSInteger hadSelectValue = 1;
                [self.dataArray[sectionIdx] enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger rowIdx, BOOL * _Nonnull rowStop) {
                    if ([self.chooseSelectArray containsObject:obj]) {
                        hadSelectValue = [obj integerValue];
                        *rowStop = YES;
                    }
                }];
                [self.dataArray[sectionIdx] enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger rowIdx, BOOL * _Nonnull rowStop) {
                    // 获取与之对应的 Cell 中的 Button
                    UIButton *button = [self getCellWithSection:sectionIdx row:rowIdx];
                    // 获取当前的 Value
                    NSInteger buttonValueInter = [self.dataArray[sectionIdx][rowIdx] integerValue];
                    // 就是没有选中
                    if (!button.selected) {
                        [self.userAvailabilityArray enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
                            button.enabled = NO;
                            if (([obj integerValue] % (allValueIntger / hadSelectValue * buttonValueInter)) == 0) {
                                button.enabled = YES;
                                *stop = YES;
                            }
                        }];
                    }
                }];

            }];
        }
     }];
}

目前逻辑基本OK啦，算法的复杂度还很高，继续优化中。。。

最后编辑于：2017.12.09 01:51:09

人面猴
序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
沈念sama阅读 206,482评论 6赞 481
死咒
序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
沈念sama阅读 88,377评论 2赞 382
救了他两次的神仙让他今天三更去死
文/潘晓璐我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
开封第一讲书人阅读 152,762评论 0赞 342
道士缉凶录：失踪的卖姜人
文/不坏的土叔我叫张陵，是天一观的道长。经常有香客问我，道长，这世上最难降的妖魔是什么？我笑而不...
开封第一讲书人阅读 55,273评论 1赞 279
港岛之恋（遗憾婚礼）
正文为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
茶点故事阅读 64,289评论 5赞 373
恶毒庶女顶嫁案：这布局不是一般人想出来的
文/花漫我一把揭开白布。她就那样静静地躺着，像睡着了一般。火红的嫁衣衬着肌肤如雪。梳的纹丝不乱的头发上，一...
开封第一讲书人阅读 49,046评论 1赞 285
城市分裂传说
那天，我揣着相机与录音，去河边找鬼。笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。我是一名探鬼主播，决...
沈念sama阅读 38,351评论 3赞 400
双鸳鸯连环套：你想象不到人心有多黑
文/苍兰香墨我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
开封第一讲书人阅读 36,988评论 0赞 259
万荣杀人案实录
序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
沈念sama阅读 43,476评论 1赞 300
护林员之死
正文独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋以下内容为张勋视角年9月15日...
茶点故事阅读 35,948评论 2赞 324
白月光启示录
正文我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
茶点故事阅读 38,064评论 1赞 333
活死人
序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
沈念sama阅读 33,712评论 4赞 323
日本核电站爆炸内幕
正文年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
茶点故事阅读 39,261评论 3赞 307
男人毒药：我在死后第九天来索命
文/蒙蒙一、第九天我趴在偏房一处隐蔽的房顶上张望。院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
开封第一讲书人阅读 30,264评论 0赞 19
一桩弑父案，背后竟有这般阴谋
文/苍兰香墨我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。一阵脚步声响...
开封第一讲书人阅读 31,486评论 1赞 262
情欲美人皮
我被黑心中介骗来泰国打工，没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。一个月前我还...
沈念sama阅读 45,511评论 2赞 354
代替公主和亲
正文我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
茶点故事阅读 42,802评论 2赞 345

iOS 中类似 SKU 刷选，利用质数实现

由于上述仍然有问题，不够严谨，在不优先考虑算法的前提下，继续优化。

改进一、逻辑优化

推荐阅读更多精彩内容