写了个for_each宏,放出来给各位评审评审。

摘要: 转载请注明出处。http://www.cnblogs.com/dave_cn/ 

之前在编码的时候遇到几次需要将这么几个值统一处理下,当时也没有想到什么好招,就每个量都写了相同的代码,一直觉得很土,加之使用python时的for...in...的美好感觉,便写了个for_each的宏。

for_each宏能够很方便遍历一组零散的元素,而且在遍历完之后将不再需要的临时申请的空间释放掉。

set_list_m每次添加一个元素,set_list_f则会一次将所需要遍历的元素全部加入。

因为懒set_list_m宏中应该给变量加括号的也就懒着加了。

如果有问题的地方,还请各位多多指正!


#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>

struct list_t {
        void                 *value;
        struct list_t         *next;
};

#define set_list_m(_p, _list) \
        do { \
                if (_list == NULL) { \
                        _list = (struct list_t *)malloc(sizeof(struct list_t)); \
                        _list->next = NULL; \
                        _list->value = _p; \
                } \
                else { \
                        struct list_t *_tmp = _list; \
                        while (_tmp->next != NULL) { \
                                _tmp = _tmp->next; \
                        } \
                        _tmp->next = (struct list_t *)malloc(sizeof(struct list_t)); \
                        _tmp = _tmp->next; \
                        _tmp->next = NULL; \
                        _tmp->value = _p; \
                } \
        } \
        while (0)

/*
 * 写for_each宏的原由是为了方便遍历一组零散的元素(这些元素并没有通过数组/列表等组织在一起)
 * 而python中的for...in...也不停诱惑着我
 */
#define for_each(_p, _list) \
        for ( struct list_t *_tmp = (_list), *_front; \
              (_tmp != NULL) && (((_p) = _tmp->value) || 1); \
              _front = _tmp, _tmp = _tmp->next, free(_front) )

/*
 * struct list_t *set_list_f(void *p, ...)
 * @parameter: 需要要以NULL结束
 * @return:    返回for_each可用的list
 */
struct list_t *set_list_f(void *p, .../* NULL */)
{
        va_list arg_ptr;
        va_start(arg_ptr, p);

        struct list_t *_tmpnode = NULL;
        struct list_t *list = NULL;
        void *_tmpval = p;
        while (_tmpval != NULL) {
                if (_tmpnode == NULL) {
                        _tmpnode = (struct list_t *)malloc(sizeof(struct list_t));
                        list = _tmpnode;
                }
                else {
                        _tmpnode->next = (struct list_t *)malloc(sizeof(struct list_t));
                        _tmpnode = _tmpnode->next;
                }
                _tmpnode->next = NULL;
                _tmpnode->value = _tmpval;

                _tmpval = va_arg(arg_ptr, void *);
        }

        va_end(arg_ptr);

        return list;
}

int main()
{
        //////////////////////////////////////////////////////
        int a = 1, b = 2, c = 3;

        struct list_t *l1 = NULL;
        set_list_m(&a, l1);
        set_list_m(&b, l1);
        set_list_m(&c, l1);

        int *p1 = NULL;
        for_each(p1, l1) {
                printf("for_each\n");
                *p1 = 4;
        }

        printf("a = [%d], b = [%d], c = [%d]\n", a, b, c);
        
        //////////////////////////////////////////////////////
        a = 1, b = 2, c = 3;

        struct list_t *l2 = set_list_f(&a, &b, &c, NULL);

        int *p2 = NULL;
        for_each(p2, l2) {
                printf("for_each\n");
                *p2 = 5;
        }

        printf("a = [%d], b = [%d], c = [%d]\n", a, b, c);
        
        return 0;
}

posted @ 2010-11-19 00:13  mr. dave  阅读(2809)  评论(3编辑  收藏  举报