URLs

题目链接: http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=243

题意: 给一个URL 找出URL的protocol、host、port、path，port和path可能不给出(即输出)。一道水题，但是一开始想法没有AC，后来改了改AC了…浪费了挺多时间

AC的代码:

#include<bits/stdc++.h>
using namespace std;
int n;
int cnt;
void solve(){
   string s;
   cin>>s;
   string protocal,host,port="<default>",path="<default>";
   int i;
   i=s.find("://");
   protocal=s.substr(0,i);
   s.erase(0,i+3);
   i=s.find("/");
   if(i!=string::npos){
        host=s.substr(0,i);
        path=s.substr(i+1,s.size());
        s.erase(i,s.size());
    }
    else{
        host=s.substr(0,s.size());
    }
   i=s.find(":");
   if(i!=string::npos){
    host=s.substr(0,i);
    s.erase(0,i+1);
    port=s;
   }
    cout<<"URL #"<<cnt<<endl;
    cout<<"Protocol = "<<protocal<<endl;
    cout<<"Host     = "<<host<<endl;
    cout<<"Port     = "<<port<<endl;
    cout<<"Path     = "<<path<<endl;
}

int main()
{
    scanf("%d",&n);
    cnt=0;
    while(n--){
        cnt++;
        solve();
        cout<<endl;
    }
    return 0;
}

WA的代码，具体原因我也不知道，感觉自己思路是对的…

#include<bits/stdc++.h>
using namespace std;
int n;
int cnt;
void solve(){
   string s;
   cin>>s;
   string protocal,host,port="<default>",path="<default>";
   int i;
   i=s.find("://");
   protocal=s.substr(0,i);
   s.erase(0,i+3);
   i=s.find(":");
   if(i!=string::npos){
    host=s.substr(0,i);
    s.erase(0,i+1);
    i=s.find("/");
    if(i!=string::npos){
        port=s.substr(0,i);
        s.erase(0,i+1);
        path=s.substr(0,s.size());
    }
    else{
        port=s.substr(0,s.size());

    }
   }
   else{
    i=s.find("/");
    if(i!=string::npos){
        host=s.substr(0,i);
        s.erase(0,i+1);
        path=s.substr(0,s.size());
    }
    else{
        host=s.substr(0,s.size());
    }
   }
    cout<<"URL #"<<cnt<<endl;
    cout<<"Protocol = "<<protocal<<endl;
    cout<<"Host     = "<<host<<endl;
    cout<<"Port     = "<<port<<endl;
    cout<<"Path     = "<<path<<endl;
}

int main()
{
    scanf("%d",&n);
    cnt=0;
    while(n--){
        cnt++;
        solve();
        cout<<endl;
    }
    return 0;
}