ZOJ1009 Enigma

#include <iostream> 
#include <string> 
using namespace std; 
 
int rotor[3][26];//密码表
int rround[3]; //加权轮转表
int totalLetter; 
 
void nextStep() 
{//密码表轮转
    rround[0]++; 
    if (rround[0]%totalLetter == 0) 
    { 
        rround[1]++; 
        rround[0] = 0; 
        if (rround[1]%totalLetter == 0) 
        { 
            rround[2]++; 
            rround[2] %= totalLetter; 
            rround[1] = 0; 
        } 
    }    
} 
 
void revolve(char ch) 
{//利用密码表解密
    int num,i; 
    num=ch-'A';    
    for(i=2;i>=0;i--) 
    {
        num+=rotor[i][(num-rround[i]+ totalLetter)%totalLetter]; 
        num=(num+totalLetter)%totalLetter; 
    }    
    cout<<(char)('a'+num); 
} 
 
int main() 
{
    string curLine;
    int i,j,nCase;    
    int numCrypt; 
    nCase=1; 
    while(cin>>totalLetter&&totalLetter!=0) 
    { 
        memset(rotor,0,sizeof(rotor)); //密码表清零
        //初始化密码表
        for(i=0;i<3;i++) 
        {
            cin>>curLine;
            for(j=0;j<totalLetter;++j) 
            { 
                rotor[i][curLine[j]-'A']=j-(curLine[j]-'A'); 
            } 
        }     
        if(nCase!=1)
            cout<<endl; 
        cin>>numCrypt;
        cout<<"Enigma "<<nCase++<<":"<<endl; 
        for(i=0;i<numCrypt;i++) 
        { 
            memset(rround,0,sizeof(rround));//加权轮转表清零
            j=0; 
            cin>>curLine;//输入密文
            while(curLine[j]!='\0') 
            { 
                revolve(curLine[j++]);//解密
                nextStep();//密码表轮转   
            } 
            cout<<endl;          
        }        
    } 
    return 0; 
}