【USACO】The Longest Prefix

嗨呀,用了std::ios::sync_with_stdio(false),比答案块一丢丢。


用了一点DP的思想。对每个字符,因为prefix最长为十的原因,只需要,检测其前缀(从长度为一到长度为十,包括该字符本身)是否在primitive集合中。

/*
ID :
LANG: C++11
TASK: prefix
 */


#include 
#include 
#include 
#include 
#include 
#include 

using namespace std;

set primitives;

int main()
{
    freopen("prefix.in", "r", stdin);
    freopen("prefix.out", "w", stdout);
    std::ios::sync_with_stdio(false);

    int ans = 0;
    int p = -1;
    string s, tail; //每次读入的字符串以及,上一行字符串的结尾10个字符,这十个字符可能和下一行的前几位相连且在primitives集合里
    while (cin >> s){
        if (s == ".")
            break;
        primitives.insert(s);
    }

    bool bOK[15] = {};  //记录,当前字符,往前的10个位置,是否是一个prefix
    bool bEnd = false;  //结束符号,如果连续十个位置都不是prefix,则结束
    while (cin >> s){   //读入测试数据
        s = tail + s;
        int i = tail.length();
        p = -1; //当前最长prefix的位置
        for (int i = 0; i <= 9; i ++){  //计算上一行字符串的tail连到当前行时,最长prefix的位置
            if (bOK[i])
                p = i;
        }
        while (s[i] != 0){
            for (int j = 1; j <= 10; j ++){ //当前位置i,如果从开头到i-j是一个prefix,且i-j到i在primitives集合中,则从开头到i也是一个prefix
                if (i - j < 0){ //处理开头前十个
                    if (primitives.find(s.substr(0, j)) != primitives.end()){
                        bOK[10] = true;
                    }
                    break;
                }
                else{
                    if (bOK[10 - j] && (primitives.find(s.substr(i - j + 1 , j)) != primitives.end())){
                        bOK[10] = true;
                        break;
                    }
                }
            }
            if (bOK[10]){
                ans += i - p;
                p = i;
            }
            bool count = false;
            for (int i = 1; i <= 10; i ++){
                if (bOK[i]){
                    count = true;
                }
                bOK[i - 1] = bOK[i];
            }
            bOK[10] = false;
            if (!count && i >= 9){
                bEnd = true;
                break;
            }
            i ++;
        }

        if (bEnd){
            break;
        }
        tail = i >= 10 ? s.substr(i - 10, 10) : s;
    }

    cout << ans << endl;

    return 0;
}

你可能感兴趣的:(DP,USACO)