Species Tree 利用HashTable实现实例代码
SpeciesTree利用HashTable实现
题目再现
题目内容:
给定一个物种演化图, 关系的表示方式如下: xy:表示x为y的先祖。 一个物种只会有一个先祖, 一个先祖可以有很多个演化出来的物种, 请你找出每个问题询问物种的祖父物种(先祖的先祖), 每个物种会使用一个不重复的编号来表示, 如果该物种没有祖父物种的话或是不存在, 那么请将他的祖父物种当是0。(凭空而生) 保证所有物种间一定有所关连, 且不会有重复演化的现象发生, 即演化图只会是一棵树。 输入格式: 只有一组测资。 第一行会有两个数字N、Q,代表总共有N个物种及Q个问题。 接下来N-1行,每一行有两个数字x、y, 意义如题目所述。 接下来的Q行,每一行有一个数字Z, 代表要询问的物种编号。 测资范围: 1<N<10000 0<Q<1000 0<x,y,z<1000000 输出格式: 对于每一个询问的物种编号,将他们的祖父物种编号加总后再输出。 输入样例: 63 10002000 10003000 20004000 30005000 50006000 5000 6000 1234 输出样例: 4000 时间限制:100ms内存限制:16000kb
算法实现
1.简单数组下标查找法
通过题目的要求,这里可以使用最简单的方法,因为输入的x,y中,y的值是确定不变的,所以这里可以定义一个y的取值范围那么大的数组,下标就是y的值,内容就是x的值,这样查找起来十分方便,时间复杂度是O(1),但是空间上就会浪费比较多了。
#include<stdio.h> #include<string.h> intmain(){ intarrBucket[1000000]; intN,Q; intx,y,z; longlongsum=0; memset(arrBucket,0,sizeof(arrBucket)); scanf("%d%d",&N,&Q); while(N-->1){ scanf("%d%d",&x,&y); arrBucket[y]=x; } while(Q--){ scanf("%d",&z); if(arrBucket[z]!=0){ if(arrBucket[arrBucket[z]]!=0){ sum+=arrBucket[arrBucket[z]]; } } } printf("%lld",sum); return0; }
2.Hash表实现
因为方法1中,浪费空间严重,所以这里使用Hash表实现。
#include<stdio.h> #include<stdlib.h> #defineNULLKEY-1 typedefstruct{ int*elem;//元素 int*elemP;//父元素 intcount; }HashTable; intHash(HashTableH,intk){ returnk%H.count; } voidInitHashTable(HashTable*H){ inti; H->elem=(int*)malloc(sizeof(int)*H->count); H->elemP=(int*)malloc(sizeof(int)*H->count); for(i=0;i<H->count;i++){ H->elem[i]=NULLKEY; H->elemP[i]=NULLKEY; } } voidInsertHash(HashTable*H,intkey,intvalue){ intaddr; addr=Hash(*H,key); while(H->elem[addr]!=NULLKEY){ addr=(addr+1)%H->count; } H->elem[addr]=key; H->elemP[addr]=value; } intFindHash(HashTable*H,intkey,int*addr){ *addr=Hash(*H,key); while(H->elem[*addr]!=key){ *addr=(*addr+1)%H->count; if(H->elem[*addr]==NULLKEY||*addr==Hash(*H,key)){ return0; } } return1; } intmain(){ intN,Q; intx,y,z,addr; longlongsum=0; HashTableH; scanf("%d%d",&N,&Q); H.count=N-1; InitHashTable(&H); while(N-->1){ scanf("%d%d",&x,&y); InsertHash(&H,y,x); } while(Q--){ scanf("%d",&z); if(FindHash(&H,z,&addr)){ if(FindHash(&H,H.elemP[addr],&addr)){ sum+=H.elemP[addr]; } } } printf("%lld",sum); return0; }
3.用C++的map来实现
看着用C实现起来,相对来说实现的各个功能都要自己写,这里看一下用C++的STL里面的map来实现上面的题目,非常简单(不得不说STL真的很好用,但是不如HashTable速度快,空间上也不如HashTable占用的小)
#include<iostream> #include<map> usingnamespacestd; intmain(){ intn,q; cin>>n>>q; map<int,int>mp; map<int,int>::iteratorit; intx,y,z; for(inti=1;i<n;++i){ cin>>x>>y; mp.insert(pair<int,int>(y,x)); } intsum=0; for(inti=0;i<q;++i){ cin>>z; it=mp.find(z); if(it!=mp.end()){ it=mp.find(it->second); if(it!=mp.end()) sum+=it->second; } } cout<<sum; return0; }
感谢阅读,希望能帮助到大家,谢谢大家对本站的支持!