习题4-9 UVA1591 Data Mining（19行AC代码）

紫书刷题进行中，题解系列点这里

习题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;
}