IP trie树接口

自己实现了一个IP trie树接口.

在这里保存一下，方便备份以后使用，同时欢迎纠错和交流，希望有大神能指教更高效的算法.

1.头文件如下(iptrie.h)

 1 #ifndef _IP_TRIE_H_
 2 #define _IP_TIRE_H_
 3
 4 #define SPLIT_SIGN "."
 5 #define IP_BINARY_LEN 32
 6
 7 typedef struct ip_trie_node
 8 {
 9     struct ip_trie_node *child[2]; //two child node
10 }ip_trie_node;
11
12 ip_trie_node *create_iptrie_node();
13
14 void insert_iptrie_node(ip_trie_node *root,char ip[]);
15
16 int select_iptrie_node(ip_trie_node *root,char ip[]);
17
18 #endif

2.c文件如下(iptrie.c)

  1 #include <stdio.h>
  2 #include <stdlib.h>
  3 #include <string.h>
  4
  5 #include "iptrie.h"
  6
  7 /*
  8  *name: itobinary
  9  *
 10  *param:
 11  * num: orignal number; binary_str: dest string; index: the binary str copy index
 12  *
 13  *return:
 14  * void
 15  */
 16 void itobinary(int num,char binary_str[],int index)
 17 {
 18     if(binary_str == NULL)
 19     {
 20         return;
 21     }
 22
 23     int i,bit = 0x01;
 24     for(i = 0; i < 8; i++)
 25     {//conver integer to 8 bit binary str
 26         if((num & bit) != 0)
 27         {//oprater & is lower than !=
 28             binary_str[index + 7 - i] = '1';
 29         }
 30         else
 31         {
 32             binary_str[index + 7 - i] = '0';
 33         }
 34
 35         bit <<= 1; //bit * 2
 36     }
 37 }
 38
 39 /*
 40  *name: convert_ip_binary
 41  *
 42  *param:
 43  * ip:orign ip string; binary_str:dest binary string
 44  *
 45  *return:
 46  * void
 47  */
 48 void convert_ip_binary(char ip[],char binary_str[])
 49 {
 50     if(ip == NULL || binary_str == NULL)
 51     {
 52         return;
 53     }
 54
 55     /*为确保正确性在进行转换之前可以进一步进行IP格式校验*/
 56
 57     char *ip_sub = NULL;
 58     int i,index =0;
 59
 60     ip_sub = strtok(ip,SPLIT_SIGN); //slit ip by .
 61
 62     itobinary(atoi(ip_sub),binary_str,index);
 63
 64     for(i = 0; i < 3; i++)
 65     {//need to ip legal detect to pretend error
 66         ip_sub = strtok(NULL,SPLIT_SIGN);
 67
 68         index += 8;
 69         itobinary(atoi(ip_sub),binary_str,index);
 70     }
 71
 72 }
 73
 74 /*
 75  *name: create_iptrie_node
 76  *
 77  *return:
 78  * new ip trie node
 79  */
 80 ip_trie_node *create_iptrie_node()
 81 {
 82     ip_trie_node *node = (ip_trie_node *)calloc(1,sizeof(ip_trie_node));
 83
 84     if(node == NULL)
 85     {
 86         perror("create ip trie node error -- calloc");
 87     }
 88     else
 89     {
 90         node->child[0] = NULL;
 91         node->child[1] = NULL;
 92     }
 93
 94     return node;
 95 }
 96
 97 /*
 98  *name: insert_iptrie_node
 99  *
100  *param:
101  * root: trie root; ip: orignal ip string
102  *
103  *return:
104  * void
105  *
106  *notice:
107  * this function call strtok it will change input ip
108  * so if input ip need to use at other position
109  * you shold input a copy of ip
110  */
111 void insert_iptrie_node(ip_trie_node *root,char ip[])
112 {
113     if(root == NULL)
114     {
115         printf("trie have not init\n");
116
117         return;
118     }
119
120     if(ip == NULL)
121     {
122         return;
123     }
124
125     char binary_str[IP_BINARY_LEN + 1];
126     int i,child_index;
127
128     memset(binary_str,0,IP_BINARY_LEN + 1);
129
130     convert_ip_binary(ip,binary_str); //to binary string
131
132     for(i = 0; i < IP_BINARY_LEN; i++)
133     {
134         child_index = binary_str[i] - '0'; //child is 0 or 1
135         if(root->child[child_index] == NULL)
136         {
137             root->child[child_index] = create_iptrie_node();
138         }
139
140         root = root->child[child_index];
141     }
142 }
143
144 /*
145  *name: select_iptrie_node
146  *
147  *param:
148  * root: trie root; ip: orignal ip string
149  *
150  *return:
151  * 0 :not find; 1:find
152  *
153  *notice:
154  * this function call strtok it will change input ip
155  * so if input ip need to use at other position
156  * you shold input a copy of ip
157  */
158 int select_iptrie_node(ip_trie_node *root,char ip[])
159 {
160     if(root == NULL)
161     {
162         printf("trie have not init\n");
163         return 0;
164     }
165
166     if(ip == NULL)
167     {
168         return 0;
169     }
170
171     int i;
172     char binary_str[IP_BINARY_LEN + 1];
173
174     memset(binary_str,0,IP_BINARY_LEN + 1);
175
176     convert_ip_binary(ip,binary_str); //to binary string
177
178     int child_index;
179     for(i = 0; i < IP_BINARY_LEN; i++)
180     {
181         child_index = binary_str[i] - '0';
182
183         if(root->child[child_index] == NULL)
184         {
185             return 0;
186         }
187
188         root = root->child[child_index];
189     }
190
191     return 1;
192 }

3.main.c如下(测试程序)

 1 #include <stdio.h>
 2 #include <stdlib.h>
 3
 4 #include "iptrie.h"
 5
 6
 7 int main()
 8 {
 9     char sip[16];
10     char dip[16];
11     int i = 0;
12     int isfind = 0;
13     ip_trie_node *root = create_iptrie_node();
14
15     while(1)
16     {
17         printf("insert a ip:\n");
18         scanf("%s",sip);
19         insert_iptrie_node(root,sip);
20
21         printf("query a ip:\n");
22         scanf("%s",dip);
23         isfind = select_iptrie_node(root,dip);
24         if(isfind == 1)
25         {
26             printf("find\n");
27         }
28         else
29         {
30             printf("not find\n");
31         }
32     }
33 }

4.Makefile (linux下编译)

CC = gcc
CFLAG = -gINC = -I./target:IptrieIptrie:iptrie.o main.c$(CC) $(CFLAG) $(INC) -o $@  $^iptrie.o:iptrie.c$(CC) -c $<clean:rm *.o Iptrie

转载于:https://www.cnblogs.com/daimadebanyungong/p/5130622.html

IP trie树接口相关推荐

trie树java_【数据结构】Trie树的应用：查询IP地址的ISP（Java实现）
查询IP地址的ISP 给定一个IP地址,如何查询其所属的ISP,如:中国移动(ChinaMobile),中国电信(ChinaTelecom),中国铁通(ChinaTietong)? 现在网上有ISP的 ...
mysql索引用trie树_数据结构与算法之美【完整版】
资源目录: ├─01-开篇词 (1讲) │ ├─00丨开篇词丨从今天起,跨过"数据结构与算法"这道坎.html │ ├─00丨开篇词丨从今天起,跨过"数据结构与算法&qu ...
数据结构之trie树——First! G，电子字典，Type Printer，Nikitosh and xor
文章目录 [USACO12DEC]First! G [JSOI2009]电子字典 [IOI2008] Type Printer Nikitosh and xor [USACO12DEC]First! ...
trie树 mysql_Trie树详解（转）
特别声明本文只是一篇笔记类的文章,所以不存在什么抄袭之类的. 以下为我研究时参考过的链接(有很多,这里我只列出我记得的): 1.字典树的概念字典树,因为它的搜索快捷的特性被单词搜索系统使用,故又称 ...
实现Trie树（C++）
文章目录前言一.Trie树原理二.Trie树实现(C++) 1.接口 2.实现总结前言 Trie树,又叫前缀树,字典树,单词查找树,是由二叉树衍生出来的一种树形高级数据结构.经常用于处理字符 ...
【图解算法】Trie树
欢迎来到我的算法专栏,今天我们来讲一种常见的数据结构:Trie树. 我们会对Trie树的性质,相关操作,应用进行讲解,并模拟实现一个简版的Trie树. 目录 1. Trie树简介 2. Trie树的 ...
usaco Cowxor (trie 树）
没想到trie树还可以用在这上面,厉害厉害. [分析]这是字母树的经典应用.首先因为是求xor的最大值,可以用前缀和计算xor值,然后n^2枚举即可. [cpp] view plaincopy for ...
字符串匹配算法 -- AC自动机基于Trie树的高效的敏感词过滤算法
文章目录 1. 算法背景 2. AC自动机实现原理 2.1 构建失败指针 2.2 依赖失败指针过滤敏感词 3. 复杂度及完整代码 1. 算法背景之前介绍过单模式串匹配的高效算法:BM和KMP 以及 ...
字符串匹配数据结构 --Trie树高效实现搜索词提示 / IDE自动补全
文章目录 1. 算法背景 2. Trie 树实现原理 2.1 Trie 树的构建 2.2 Trie树的查找 2.3 Trie树的遍历 2.4 Trie树的时间/空间复杂度 2.5 Trie 树 Vs ...

IP trie树接口

IP trie树接口相关推荐

最新文章

热门文章