USACO算法系列六——contact

    题目：http://www.nocow.cn/index.php/Translate:USACO/contact

    题目很长，又是奶牛，又是望眼镜，还有地外生物的。很佩服外国人的想象力，一道算法题也能联系出一部星际大战出来。

    题目大意，简而言之，就是统计由01组成的字符串当中，长度在A到B之间（1<=A<B<=12）的字符串出现的频率，然后把前N个最大频率的字符串打印出来。

    看起来是简洁多了，但是题目还是比较复杂的。想到了两种方案：

    第一种是把01串看作是一个个位。AB长度在12之内，因此用int二维数组表示出现的频率。int [number][length]，number表示这个位串组成的值，length表示位串的长度，因为000的值是0，00的值也是0。所以用一个length加以区分。然后开始遍历这个串，靠不断的移位和更新二维数组统计词频。

    第二种方案是把01字符串看作是一个二叉树到达01路径。然后用一个满二叉树数组，长度为2^(B+1)-1来存储各个节点出现的频率。用递归不断移动和匹配字符串。‘0’往左子节点空间为2 * index，‘1’右节点空间为 2 * index + 1。时间复杂度为o（（B-A）*字符串长度）。

     我用第二种方案实现的代码如下：

#include <iostream> #include <fstream> #include <math.h> #define LENGTH 200000 #define MAX 8192 using namespace std; int a,b,n; int tree[MAX]; char letter[LENGTH]; int length = 0; ifstream fin("contact.in"); ofstream fout("contact.out"); void contact(int index,int layer,int position) { if (layer > b || index >= length) { return; } else { if (letter[index] == '0') { position = position * 2 + 1; } else { position = position * 2 + 2; } if (layer >= a) { tree[position] ++; }//在范围内 contact(index+1,layer+1,position); } } //打印结果 void printResult(int num) { int l = 0; int result[12]; while (num != 1) { if ((num&1) == 1) { result[l] = 1; } else { result[l] = 0; } l ++; num = num >> 1; } for (int i =l-1; i >=0; i --) { fout << result[i]; } } int main() { fin >> a >> b >> n; while (!fin.eof()) { fin >> letter[length]; length ++; } length --; //遍历数组 for (int i=0; i < length; i ++) { contact(i,1,0); } //打印结果 int pl = pow((double)2,b+1) -1; int start = pow((double)2,a) -1; for (int i = 0; i < n; i ++ ) { int max = 0; for (int j = start; j < pl; j ++) { if (max < tree[j]) { max = tree[j]; }//end if } if (max == 0) { return 0; } fout << max; int number = 0; for (int k=start; k < pl; k ++) { if (tree[k] == max) { if (number%6 == 0) { fout << endl; } else fout<<" "; printResult(k+1); number ++; tree[k] = -1; } } fout << endl; } return 0; }

    顺利通过了USACO的测试，结果如下：

Executing... Test 1: TEST OK [0.011 secs, 3240 KB] Test 2: TEST OK [0.000 secs, 3240 KB] Test 3: TEST OK [0.011 secs, 3240 KB] Test 4: TEST OK [0.011 secs, 3240 KB] Test 5: TEST OK [0.032 secs, 3240 KB] Test 6: TEST OK [0.032 secs, 3240 KB] Test 7: TEST OK [0.076 secs, 3240 KB]

   

     最后感觉USACO对空格的要求太严格了，而空格又在输出要求中看不到，所以每次都没办法一次AC。