紫书刷题进行中,题解系列点这里
习题4-9 UVA1591 Data Mining(19行AC代码)
思路分析
纸老虎题目,看着很难,其实读懂题目后很简单
题目背景就是为了提高访问数组内容的时间和空间效率,只需找到使K最小的A和B,若存在多解,则按KAB顺序取字典序最小者。
时间优化(乘除变加减)
假设有P和Q两个结构体数组,各自记录的大小分别为Sp和Sq。通常访问第i个元素是通过如下乘法实现,但在计算机中加减法效率比乘除法高许多,因此可将其转换为加减法运算来访问
P[i]=Sp*i // 乘法
Q[i]=Sq*i // 乘法
P[i]=P[i-1]+Sp // 加法
P[i]=P[i+1]-Sp // 减法
然而Q转换稍微复杂些,需按如下公式计算,同样达到时间效率提高的目的
Q(i) = (P(i) + P(i) << A) >> B //公式(1)
空间优化
假设确定A和B,数组有N个元素(0~N-1),那么数组所需空间:
K=Q(n-1)+Sq // 第n-1个元素偏移+一个记录大小
因此,不同A和B对应的K不同,由于题目说明整数范围内可解决,所以A,B取值范围:[0,31]
(32位最多移位31),只要遍历32*32种可能,选出最小K对应的A,B即可
- 如何判断A和B合法?
由公式(1)可知Q(i)为等差数列,只要公差d>=Sq,就能够存储数组Q,因此,推导可得:
d=Q(i+1)-Q(i)=(P(i+1)-P(i))(1+2^A)/(2^B)=Sp(1+2^A)/(2^B) >= Sq
//为了计算过程不溢出且避免除法误差,变换不等式如下
Sp+(Sp<<A) >= Sq<<B // 注意C语言中移位优先级低于四则运算
注意点
- 注意整型移位范围在[0,31],可能存在溢出,可用long long类型避免该问题
- 判断AB是否合法时注意变换不等式,避免除法和乘法溢出
- 移位优先级低于四则运算,注意使用括号
AC代码(C++11)
直接比较版(19行代码)
#include<bits/stdc++.h>
using namespace std;
typedef long long LL; // 避免移位溢出
int main() {
LL n, Sp, Sq;
while (cin >>n >>Sp >>Sq) {
LL K=LONG_LONG_MAX, A, B;
for (int i = 0; i < 32; i ++) { // A
for (int j = 0; j < 32; j ++) { // B
if (((Sp + (Sp << i)) >= (Sq << j)) && (K > ((Sp*(n-1)+((Sp*(n-1))<<i))>>j)+Sq)) { // 公差大于Sq且k更大
K = ((Sp*(n-1)+((Sp*(n-1))<<i))>>j)+Sq; // 更新KAB
A = i; B = j;
}
}
}
printf("%lld %lld %lld\n", K, A, B); // 第一个即符合条件
}
return 0;
}
sort排序版(28行代码)
#include<bits/stdc++.h>
using namespace std;
typedef long long LL; // 避免移位溢出
typedef struct KAB {
LL K, A, B;
KAB(LL _k, LL _A, LL _B) : K(_k), A(_A), B(_B){} // 默认构造
}KAB;
bool cmp(const KAB& a, const KAB& b) { // 按KAB顺序取字典序小者
if (a.K != b.K) return a.K < b.K; // 取K小
else if (a.A != b.A) return a.A < b.A; // 取A小
else return a.B < b.B; // 取B小
}
LL n, Sp, Sq;
int main() {
while (cin >>n >>Sp >>Sq) {
vector<KAB> kab; // 存储所有可能值
for (int i = 0; i < 32; i ++) { // A
for (int j = 0; j < 32; j ++) { // B
if ((Sp + (Sp << i)) >= (Sq << j)) { // 公差大于Sq
kab.push_back({((Sp*(n-1)+((Sp*(n-1))<<i))>>j)+Sq, i, j}); // 移位优先级低于=-*/
}
}
}
sort(kab.begin(), kab.end(), cmp); // 排序
printf("%lld %lld %lld\n", kab[0].K, kab[0].A, kab[0].B); // 第一个即符合条件
}
return 0;
}