代码练习

选举投票小练习

#include <iostream>
#include <string>
using namespace std;



 class Node
{
public:
    Node(){}
    Node(string name){candidateName = name;next=NULL;}
    void setNext(Node *p){next=p;}
    void setTicketNumber(int n){ticketNumber=n;}
    void setCandidateName(string name){candidateName = name;}
    int getTicketNumber(){return ticketNumber;}
    string getCandidateName(){return candidateName;}
    Node *getNext(){return next;}
    
private:
    int ticketNumber;
    string candidateName;
    Node *next;
    
};

class Linklist //链表类
{
public:
    Linklist(){head=NULL;}
    Linklist(string name){ head = new Node(name); }
    void addAtEnd(string name);
    void visitAllNode();
    int voteCandidate(string name);

private:
    Node *head;
};
void Linklist::addAtEnd(string name)  //往链表尾添加节点
{
    Node *p=head;
    if(head==NULL){ head = new Node(name);}
    else
    {
        while(p->getNext()!=NULL)
        {
            p= p->getNext();
        }//用p来指向最后一个节点
        p->setNext(new Node(name));//插入节点
        
    }
    
}

void Linklist::visitAllNode()//遍历链表
{
    Node *p;
    p=head;
    if(p==NULL){cout<<"空链表!"<<endl;}
    else
    {
        while(p->getNext()!=NULL)
        {
            p=p->getNext();
            cout<< p->getCandidateName() << " " << p->getTicketNumber() << endl;
        }
        cout << head->getTicketNumber() << endl;
    }
}

int Linklist::voteCandidate(string name)
{
    Node *p;
    p = head;
    if (p==NULL)
    {
        return 0;
    }
    else
    {
        while (p->getNext()!=NULL && p->getCandidateName()!= name)
        {
            p=p->getNext();
        }
        if(p->getCandidateName() == name)
        {
            int currentTicketNumber = p->getTicketNumber();
            p->setTicketNumber(currentTicketNumber+1);
        }
        else
        {
            head->setTicketNumber(head->getTicketNumber()+1);
        }
        
    }
    return p->getTicketNumber();
}



int main(int argc, const char * argv[]){

    Linklist *pl = NULL;
    
    pl = new Linklist("error"); //创建链表首节点为error,记录无效票
    string line;
    
    while (cin>>line)
    {
        if (line == "getVoteResult")
        {
            pl->visitAllNode();
            return 0;
        }
        else if(line == "addCandidate")
        {
            cin >> line;
            string candidateName = line;
            pl->addAtEnd(candidateName);
            
        }
        else if(line == "vote")
        {
            cin >> line;
            string candidateName = line;
            pl->voteCandidate(line);
        }
    }
    return 0;
}