iOS tableView分组

//

//  ViewController.m

//  分组tableview练习

//

//  Created by 李江 on 16/4/28.

//  Copyright © 2016年 李江. All rights reserved.

//

#import "ViewController.h"

#import "Person.h"

@interface ViewController ()<UITableViewDelegate,UITableViewDataSource>

{

    NSMutableDictionary *dict;

    NSArray *sectionList;

    NSArray *rowList;

}

@property(nonatomic,strong)UITableView *tableview;

@end

@implementation ViewController

- (void)viewDidLoad {

    [super viewDidLoad];

    

    sectionList = @[@"好友",@"同学",@"家人",@"小包"];

    rowList = @[@"大宝",@"小包",@"小哈",@"莫伸"];

    self.tableview = [[UITableView alloc]initWithFrame:self.view.frame style:UITableViewStyleGrouped];

    self.tableview.delegate =self;

    self.tableview.dataSource =self;

    [self.view addSubview:self.tableview];

   

}

- (NSInteger)numberOfSectionsInTableView:(UITableView *)tableView{

    

    return sectionList.count;

}

- (nullable UIView *)tableView:(UITableView *)tableView viewForHeaderInSection:(NSInteger)section{

    UIButton *button = [[UIButton alloc]initWithFrame:CGRectMake(0, 0, 300, 50)];

    [button setTitle:sectionList[section] forState:UIControlStateNormal];

    [button setTitleColor:[UIColor blackColor] forState:UIControlStateNormal];

    button.tag = section;

    [button addTarget:self action:@selector(buttona:) forControlEvents:UIControlEventTouchUpInside];

    return button;

}

-(void)buttona:(UIButton *)sender{

    if (!dict) {

        dict = [NSMutableDictionary dictionary];

    }

    if (![dict objectForKey:[NSString stringWithFormat:@"%ld",sender.tag]]) {

        [dict setObject:@"1" forKey:[NSString stringWithFormat:@"%ld",sender.tag]];

    }else{

        [dict removeAllObjects];

    }

    [self.tableview reloadSections:[NSIndexSet indexSetWithIndex:sender.tag] withRowAnimation:UITableViewRowAnimationFade];

}

- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section{

    if ([dict objectForKey:[NSString stringWithFormat:@"%ld",section]]) {

        return rowList.count;

    }

    else{return 0;}

  

}

- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath{

   static NSString *idd = @"ad";

    UITableViewCell *cell = [tableView dequeueReusableCellWithIdentifier:idd];

    if (!cell) {

        cell = [[UITableViewCell alloc]initWithStyle:UITableViewCellStyleSubtitle reuseIdentifier:idd];

    }

    cell.textLabel.text = rowList[indexPath.row];

    cell.textLabel.textColor = [UIColor blackColor];

    return cell;

}

- (void)didReceiveMemoryWarning {

    [super didReceiveMemoryWarning];

    // Dispose of any resources that can be recreated.

}

@end