一个简单的交通查询系统

#include "comment.h" int message::tokstr(char* dis,char **str) { int i = 0; char* temp = NULL; temp = *str; while(**str != '@') { i ++; *str = *str + 1; } memcpy(dis,temp,i); *str = *str + 1; return 0; } int message::decode(char *msg_str) { int cmd,i; char cmd_str[5]; char start_str[50]; char end_str[50]; char way_str[5]; char choose_str[5]; char cityname[20]; char trainid_str[10]; char starttime[10]; char endtime[10]; char money[10]; char cityp_str[5]; char cityid_str[5]; char strc[50]; char user_str[50]; char password_str[50]; char *strtemp; memset(cmd_str,'\0',5); strcpy(strc,msg_str); strtemp = strc; tokstr(cmd_str,&strtemp); cmd = atoi(cmd_str); this->peple = cmd; switch(cmd) { case USER: memset(start_str,0,50); tokstr(start_str,&strtemp); for ( i = 0 ; i < City.shownum();i ++) { if (strcmp(start_str,City.ShowCityInfo(i).GetName().c_str()) == 0) break; } this->require.start = i; memset(end_str,0,50); tokstr(end_str,&strtemp); for ( i = 0 ; i < City.shownum();i ++) { if (strcmp(end_str,City.ShowCityInfo(i).GetName().c_str()) == 0) break; } this->require.end = i; memset(way_str,0,5); tokstr(way_str,&strtemp); this->require.way = atoi(way_str); memset(choose_str,0,5); tokstr(choose_str,&strtemp); this->require.choose = atoi(choose_str); return 0; case MANAGE: memset(cmd_str,'\0',5); tokstr(cmd_str,&strtemp); cmd = atoi(cmd_str); this->cmd = cmd; switch(cmd) { case LOGIN: memset(user_str,'\0',50); memset(password_str,'\0',50); tokstr(user_str,&strtemp); tokstr(password_str,&strtemp); strcpy(loginuser.name,user_str); strcpy(loginuser.password,password_str); return 0; case ADDCITY: memset(cityname,'\0',20); tokstr(cityname,&strtemp); this->city_.SetName(cityname); memset(cityp_str,'\0',5); tokstr(cityp_str,&strtemp); this->city_.SetCityP(atoi(cityp_str)); memset(cityid_str,'\0',5); tokstr(cityid_str,&strtemp); this->city_.SetID(atoi(cityid_str)); return 0; case LOOKCITY: memset(cityid_str,'\0',5); tokstr(cityid_str,&strtemp); this->city_.SetID(atoi(cityid_str)); return 0; case DELETECITY: memset(cityid_str,'\0',5); tokstr(cityid_str,&strtemp); this->city_.SetID(atoi(cityid_str)); return 0; case ADDTRAIN: this->traffic_ = new TrainInfo; memset(trainid_str,'\0',5); tokstr(trainid_str,&strtemp); this->traffic_->ID= atoi(trainid_str); memset(start_str,0,5); tokstr(start_str,&strtemp);; this->traffic_->Start = atoi(start_str); memset(end_str,0,5); tokstr(end_str,&strtemp); this->traffic_->End = atoi(end_str); memset(starttime,0,10); tokstr(starttime,&strtemp); this->traffic_->StartTime = atoi(starttime); memset(endtime,0,10); tokstr(endtime,&strtemp); this->traffic_->EndTime = atoi(endtime); memset(money,0,10); tokstr(money,&strtemp); this->traffic_->Money = atoi(money); return 0; case LOOKTRAIN: this->traffic_ = new TrainInfo; memset(trainid_str,'\0',5); tokstr(trainid_str,&strtemp); this->traffic_->ID = atoi(trainid_str); return 0; case DELETETRAIN: this->traffic_ = new TrainInfo; memset(trainid_str,'\0',5); tokstr(trainid_str,&strtemp); this->traffic_->ID = atoi(trainid_str); return 0; case ADDBUS: this->traffic_ = new BusInfo; memset(trainid_str,'\0',5); tokstr(trainid_str,&strtemp); this->traffic_->ID= atoi(trainid_str); memset(start_str,0,5); tokstr(start_str,&strtemp);; this->traffic_->Start = atoi(start_str); memset(end_str,0,5); tokstr(end_str,&strtemp); this->traffic_->End = atoi(end_str); memset(starttime,0,10); tokstr(starttime,&strtemp); this->traffic_->StartTime = atoi(starttime); memset(endtime,0,10); tokstr(endtime,&strtemp); this->traffic_->EndTime = atoi(endtime); memset(money,0,10); tokstr(money,&strtemp); this->traffic_->Money = atoi(money); return 0; case LOOKBUS: this->traffic_ = new BusInfo; memset(trainid_str,'\0',5); tokstr(trainid_str,&strtemp); this->traffic_->ID = atoi(trainid_str); return 0; case DELETEBUS: this->traffic_ = new BusInfo; memset(trainid_str,'\0',5); tokstr(trainid_str,&strtemp); this->traffic_->ID = atoi(trainid_str); return 0; case LOGOFF: memset(user_str,'\0',50); tokstr(user_str,&strtemp); strcpy(loginuser.name,user_str); return 0; default: cout <<"no cmd"<< endl; return -2; } default: cout <<"no this user"<< endl; return -1; } } int message::encode() { char temp[100]; string txt1,txt2; switch(this->peple) { case USER: strcpy(buffer,this->res_.best); if (this->require.choose == MONEY) { sprintf(temp,"all money is %d\n",this->res_.money); } else { sprintf(temp,"all time is %d\n",this->res_.time); } strcat(buffer,temp); return 0; case MANAGE: if (flag == 1) { strcat(buffer," 操作OK\n"); return 0; } else if(flag == 2) { strcat(buffer,"操作NO\n"); return 0; } else { switch(cmd) { memset(buffer,0,sizeof(buffer)); case LOOKCITY: if(city_.GetCityP() == 1) txt1 = "市"; else txt1 = "县"; sprintf(buffer,"城市名：%s\n级别：%s\n城市编号:%d",city_.GetName().c_str(),txt1.c_str(),city_.GetID()); return 0; case LOOKTRAIN: txt1.clear(); txt1 = City.ShowCityInfo(traffic_->Start).GetName(); txt2 = City.ShowCityInfo(traffic_->End).GetName(); sprintf(buffer,"编号:%d\n起点站:%s\n终点站:%s\n发车时间:%d\n到站时间:%d\n票价:%d",traffic_->ID,txt1.c_str(),txt2.c_str(),traffic_->StartTime,traffic_->EndTime,traffic_->Money); return 0; case LOOKBUS: txt1.clear(); txt2.clear(); txt1 = City.ShowCityInfo(traffic_->Start).GetName(); txt2 = City.ShowCityInfo(traffic_->End).GetName(); sprintf(buffer,"编号:%d\n起点站:%s\n终点站:%s\n发车时间:%d\n到站时间:%d\n票价:%d",traffic_->ID,txt1.c_str(),txt2.c_str(),traffic_->StartTime,traffic_->EndTime,traffic_->Money); return 0; default: return -1; } } default: return -1; } return -1; } int message::process() { int num = City.shownum(); int **mat; TrafficInfo mytrain_; int j,i,weight = MAX,u; int k = 0; int wait = 0; mat = new int * [num]; for (i = 0;i <num; i++) if ((mat[i] = new int [num]) == NULL) return -1; for (i = 0;i <num; i++) for (j = 0;j < num ;j++) mat[i][j] = MAX; for (i = 0; i<num;i ++) { for (k = 0; k < Traffic.shownum(); k ++) { mytrain_ = Traffic.ShowTrafficInfo(k); if (mytrain_.Start == i) { j = mytrain_.End; switch(this->require.choose) { case MONEY: weight = mytrain_.Money; mat[i][j] = weight; break; case TIME: weight = mytrain_.EndTime - mytrain_.StartTime; mat[i][j] = weight; break; default: break; } } } } if (Dijkstra(mat) == -1) return -1; cout <<"pass "<<this->res_.best<<endl; if (this->require.choose == MONEY) cout <<"time "<<this->res_.money<<endl; else cout <<"cost"<<this->res_.time<<endl; for (i = 0 ; i< num; i ++) delete [] mat[i]; delete [] mat; return 0; } int message::Dijkstra(int** mat) { int start,end,way,choose; int num = City.shownum(); int *flag = new int [num]; int mindis,dis; int i,j,u,k,m,s; int *Distance = new int [num]; int *pass; pass = new int [num]; for (i = 0;i <num; i++) pass[i] = -1; start = this->require.start; end = this->require.end; way = this->require.way; choose = this->require.choose; for (i = 0;i < num; i++) { Distance[i] = mat[start][i]; flag[i] = 0; } flag[start] = 1; m = 0; pass[m] = start; m = 1; for (i = 0