【codevs1051】接龙游戏 栈

题目描述 Description

给出了N个单词，已经按长度排好了序。如果某单词i是某单词j的前缀，i->j算一次接龙(两个相同的单词不能算接龙)。

你的任务是：对于输入的单词，找出最长的龙。

输入描述 Input Description

第一行为N(1<=N<=105)。以下N行每行一个单词（由小写组成），已经按长度排序。(每个单词长度<50)

输出描述 Output Description

仅一个数，为最长的龙的长度。

样例输入 Sample Input

5

i

a

int

able

inter

样例输出 Sample Output

3

数据范围及提示 Data Size & Hint

1<=N<=105

栈练习。
按字典序排序，此时单词前缀相同的必定排在一起。
用一个栈来维护，若栈顶元素是当前单词的前缀（注意不能相等），则把当前单词扔进栈里，若不是前缀，则弹出，重复下去。直到栈为空或者栈顶元素是当前单词的前缀，则扔进去。在操作的同时更新答案。

代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<stack>
using namespace std;

stack<string> s;
string str[100010];

bool check(string s,string ss)
{
    if(s==ss) return false;
    for(int i=0;i<s.length();i++)
    {
        if(s[i]!=ss[i]) return false;
    }
    return true;
}

int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++) cin>>str[i];
    sort(str+1,str+1+n);
    int ans=0;
    for(int i=1;i<=n;i++)
    {
        if(s.empty()) s.push(str[i]);
        else 
        {
            while(s.size())
            {
                if(check(s.top(),str[i]))
                {
                    s.push(str[i]);
                    break;
                }
                else s.pop();
            }
            if(s.empty()) s.push(str[i]);
        }
        ans=max(ans,(int)s.size());
    }
    printf("%d",ans);
    return 0;
}