首页 > 代码库 > POJ 2418 Hardwood Species( AVL-Tree )
POJ 2418 Hardwood Species( AVL-Tree )
#include <stdio.h> #include <stdlib.h> #include <math.h> #include <string.h> typedef struct AVLTree{ char name[31]; int nCount; int nHeight; struct AVLTree* pLeft; struct AVLTree* pRight; }AVLTree; int Max( int a, int b ); int Height( AVLTree* pNode ); AVLTree* Insert( char* s, AVLTree* pNode ); AVLTree* LLRotate( AVLTree* pNode ); AVLTree* RRRotate( AVLTree* pNode ); AVLTree* LRRotate( AVLTree* pNode ); AVLTree* RLRotate( AVLTree* pNode ); void PrintTree( AVLTree* pNode ); int n = 0; int main(){ char s[31]; AVLTree* pRoot = NULL; while( gets( s ) != NULL ){ pRoot = Insert( s, pRoot ); n++; } PrintTree( pRoot ); return 0; } int Max( int a, int b ){ return ( a > b ) ? a : b; } int Height( AVLTree* pNode ){ if( pNode == NULL ) return -1; return pNode->nHeight; } AVLTree* Insert( char* s, AVLTree* pNode ){ if( pNode == NULL ){ pNode = ( AVLTree* ) malloc( sizeof( AVLTree ) ); strcpy( pNode->name, s ); pNode->nCount = 1; pNode->nHeight = 0; pNode->pLeft = NULL; pNode->pRight = NULL; } else if( strcmp( s, pNode->name ) == 0 ){ pNode->nCount++; } else if( strcmp( s, pNode->name ) < 0 ){ pNode->pLeft = Insert( s, pNode->pLeft ); if( Height( pNode->pLeft ) - Height( pNode->pRight ) == 2 ){ if( strcmp( s, pNode->pLeft->name ) < 0 ){ pNode = LLRotate( pNode ); } else{ pNode = LRRotate( pNode ); } } } else if( strcmp( s, pNode->name ) > 0 ){ pNode->pRight = Insert( s, pNode->pRight ); if( Height( pNode->pRight ) - Height( pNode->pLeft ) == 2 ){ if( strcmp( s, pNode->pRight->name ) > 0 ){ pNode = RRRotate( pNode ); } else{ pNode = RLRotate( pNode ); } } } pNode->nHeight = Max( Height( pNode->pLeft ), Height( pNode->pRight ) ) + 1; return pNode; } AVLTree* LLRotate( AVLTree* pNode ){ AVLTree* pNodeLeft = pNode->pLeft; pNode->pLeft = pNodeLeft->pRight; pNodeLeft->pRight = pNode; pNode->nHeight = Max( Height( pNode->pLeft ), Height( pNode->pRight ) ) + 1; pNodeLeft->nHeight = Max( Height( pNodeLeft->pLeft ), pNode->nHeight ) + 1; return pNodeLeft; } AVLTree* RRRotate( AVLTree* pNode ){ AVLTree* pNodeRight = pNode->pRight; pNode->pRight = pNodeRight->pLeft; pNodeRight->pLeft = pNode; pNode->nHeight = Max( Height( pNode->pLeft ), Height( pNode->pRight ) ) + 1; pNodeRight->nHeight = Max( Height( pNodeRight->pRight ), pNode->nHeight ) + 1; return pNodeRight; } AVLTree* LRRotate( AVLTree* pNode ){ pNode->pLeft = RRRotate( pNode->pLeft ); return LLRotate( pNode ); } AVLTree* RLRotate( AVLTree* pNode ){ pNode->pRight = LLRotate( pNode->pRight ); return RRRotate( pNode ); } void PrintTree( AVLTree* pRoot ){ if( pRoot == NULL ) return; PrintTree( pRoot->pLeft ); printf( "%s %.4f\n", pRoot->name, ( ( double )( pRoot->nCount ) / ( double )n ) * 100 ); PrintTree( pRoot->pRight ); }
POJ 2418 Hardwood Species( AVL-Tree )
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。