题目描述:
RGB三种颜色的球排成一行,首先消除连续相同颜色最长的最左边的一段,剩下的球左右连接起来继续以上消除过程。打印每次消除小球的颜色和编号。n<=10^6
分析:
PKU的解题报告上说用并查集+优先队列。
我不知道并查集怎么用比较合适,我用了链表存储序列+堆查找最长序列。总之代码比较繁琐,这里不想贴出来了。
写完之后人肉测试没有发现什么问题,交上去过后得到RE。我想是不是数据大了之后会有问题,于是生成一组长度为n的“单色”球序列。我发现当n>10000的时候本机上就会出错,数据量小的时候就没问题。最后发现是动态申请的链表导致的!
教训1:动态申请内存不能太多,如果需要可以在堆上开大数组用作内存池模拟内存申请。
修正之后提交结果返回了WA……
我又写了一个暴力算法的程序,然后随机生成了一些数据对比两个程序的输出结果,很多次都是完全一样。郁闷到快不行的时候海量的输出数据当中突然出现了一小段不一样。天呐,估计是边界地方有漏洞。人肉检测之后发现是调整堆的问题。
在堆中删除一个元素,通常我的做法是把最后一个元素取过来替换掉要删除的元素,然后向下调整。这样做是有漏洞的。
教训2:删除一个元素之后,除了向下调整堆,还要向上调整。
原因是有可能最后一个元素比删除的元素大,也可能比它小。堆的最后一个元素不一定是最小(或最大)的。这个和删除堆顶的操作不同。