题目描述
题目思路
我第一眼看到这个题目感觉好简单,不就是分治算法吗,直接得出答案。(分治思想变量的设定十分重要!)
然而一提交上去就是TLE!仔细一看,k 可以取到30, 直接超时。
然后翻看紫书中的内容,恍然大悟(搞了有蛮长时间!),要的只是个数字, 有一些没有必要的递归就不必执行,直接得出答案。(优化的精髓就是把不必要的重复的过程给去掉。)
题目是要求某两行之间的红色气球数量,那么涉及到区间问题了,想得到某段的结果,直接大的结果减小的结果就行。
f(k, i)
表达的意思是k小时第i行以上的红气球数量的总和。那么结果便是f(k, b) - f(k, a - 1)
接下来就是来优化暴力分治(暴力分治:最朴素的做法,划分为a、b、c三个块,第四块蓝色d那块不用管。):
---------
| a | b |
|-------|
| c | d |
---------
if i的区域包括了 a、b的区域 那么我们就可以直接把 a、b区域的红球数量加上去,再加上剩余包含c一点点的数量。
else 说明还在a、b之间,那么就直接算a中剩余的数量,然后 * 2 即可。
k小时包含多少红色气球是非常容推的,只要把题目描述的那个图做个数据记录,就得出了结果,便是3^k
。
代码实现
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
LL c(int k)
{
return k ? c(k - 1) * 3 : 1;
}
LL f(int k, int i)
{
if (!i) return 0;
if (!k) return 1;
int k2 = 1 << (k - 1);
if (i >= k2) return f(k - 1, i - k2) + c(k - 1) * 2;
return f(k - 1, i) * 2;
}
int main()
{
#ifdef LOCAL
freopen("data.in", "r", stdin);
#endif
int T;
scanf("%d", &T);
for(int kase = 1; kase <= T; ++kase)
{
int k, a, b;
scanf("%d%d%d", &k, &a, &b);
printf("Case %d: %lld\n", kase, f(k, b) - f(k, a - 1));
}
return 0;
}