USACO Contact IOI’98 TLE

毕设的东西做不下了，做点其他东西调节一下，翻出USACO Trainning继续做，到Contact这道题了，光荣地挂在了第7组数据了，写的稍微有点暴力，过段时间再把它过掉吧。

题意是这样的，有一个01串，长度200,000，有1<=A,B<=12,1<=N<=50，求的是串中长度在A,B之间的子串出现次数的统计问题。

想了个最糟糕情况为12*200000=2.4*10^6的算法，最暴力的方法，遍历所有的子串，主要是要怎么hash一个子串，将一个子串唯一映射成一个整数，0011和11是不同的子串，想了个比较巧妙的方法，把空串想为字符0，于是0011可以用1122来表示11用1122来表示，即可以压缩成一个三进制数，然后就可以方便实现了。

PS：题目好烦人，输入拆成若干行，输出又乱拆。

TLE：遍历子串有点暴力，可以尝试下记录之前的信息来递归，小小优化一下到LEN*(B-A)。

TLE的代码如下：

/*

ID: litstrong

PROG: contact

LANG: C++

*/

#include <iostream>

#include <string>

#include <algorithm>

#include <vector>

#include <set>

#include <stdio.h>

using namespace std;

const int MAX = 600005;

const int LEN = 200005;

int A, B, N;

char ch[LEN];

class CNODE

{

public:    

    string num;

    int cnt;

public:

    CNODE () : cnt(0) {}

    void AddCnt() { cnt++; }

    bool operator < (const CNODE &t) const

    {

        if(cnt != t.cnt)  return cnt > t.cnt;

        if(num.length() != t.num.length())  

            return num.length() < t.num.length();

        return num < t.num;

    }

}node[MAX];

int main()

{

    freopen("contact.in", "r", stdin);

    freopen("contact.out", "w", stdout);

    string mo = "";

    scanf("%d%d%d", &A, &B, &N);

    while(scanf("%s", ch) != EOF)  mo += ch;

    int len = mo.length();

    int len_max = 0;

    for(int i = 0; i <= len - A; i++)

    {

        string dq = "";

        int tmp = 0, thr = 1;

        for(int j = 0; j < B; j++) 

        {

            dq.push_back(mo[i + j]);

            tmp += (mo[i + j] - '0' + 1) * thr;

            thr *= 3;

            if(j < A - 1)  continue;

            if(i + j >= len)  break;

            node[tmp].num = dq;

            node[tmp].AddCnt();

            len_max = max(len_max, tmp);

        }

    }

    sort(node, node + len_max + 1);

    int before = 0, in = 0, first = 0, dqCnt = 1;

    int zNum = 0;

    for(int i = 0; i <= len_max; i++)

    {

        if(node[i].cnt)

        {

            if(node[i].cnt != before && first)  

            {

                if(++dqCnt > N)  break;

                printf("\n");

                in = 0;

                zNum = 0;

            }

            first = 1;

            if(in == 0 && zNum == 0)  printf("%d\n", node[i].cnt);

            if(in == 0)  cout << node[i].num;

            else  cout << " " << node[i].num;

            in = 1;    

            zNum++;

            if(zNum % 6 == 0) 

            {

                in = 0;

                printf("\n"); 

            }

            before = node[i].cnt;            

        }

    }

    printf("\n");

}