【算法】第四课学习笔记

一、求局部最大值

1. 题目

给定一个无重复元素的数组A[0…N-1],求找到一个该数组的局部最大值。规定:在数组边界外的值无穷小。即:A[0]>A[-1],A[N-1] >A[N]。

显然,遍历一遍可以找到全局最大值,而全局最大值显然是局部最大值,时间复杂度为O(N)。

可否有更快的办法?

2. 分析

首先给出“高原数组”定义(邹博老师自创的)。

定义:若子数组Array[from,…,to]满足

  • Array[from]>Array[from-1]
  • Array[to]>Array[to+1]

称该子数组为“高原数组”。

例如对于数组3,5,1,2,6,7,14,因为2<6,14>4,可以认为6,7,14就是其中的一个高原数组。
用图像形象地表示为:

若高原数组长度为1,则该高原数组的元素为局部最大值。

因此寻找局部最小值的算法就变成了寻找长度为1的高原数组。
(这里特别注意!题中只要求找到一个局部最大值即可!)

算法描述:
使用索引left、right分别指向数组首尾,根据定义,该数组为高原数组。

  • 求中点mid=(left+right)/2
  • A[mid]>A[mid+1],子数组A[left…mid]为高原数组
  • 丢弃后半段:right=mid
  • A[mid+1]>A[mid],子数组A[mid…right]高原数组
  • 丢弃前半段:left=mid+1
  • 递归直至left==right

该算法的时间复杂度为O(logN),最重要的是给了我们一些思维上的启发,即特别大的数据,我并不需要完全遍历一遍,就能知道这个数据中的某些有效的局部特征。

3.代码

/**
 * 寻找一个局部最小值
 */
int local_maximum(const int* a, int n){
    int left = 0;
    int right = n - 1;
    int mid;
    while(left < right){
        mid = (right - left) / 2 + left; 
        // 防溢出,等价于mid = (left + right) / 2 
        cout< a[mid+1])
            right = mid;
        else
            left = mid + 1;
    }
    return a[left];
}

二、子集和数问题

1. 题目

已知数组A[0…N-1],给定某数值sum,找出数组中的若干个数(可以为0个或n个),使得这些数的和为sum。
(假定数组中的元素都大于0:A[i]>0)

例如,
对于数组1,2,3,4,5,给定sum=10,则满足条件的若干个数有:
1, 2, 3, 4
1, 4, 5
2, 3, 5

2. 分析与代码

显然,这个问题是一个NP完全问题。

针对此题,可以设置布尔向量x[0…N-1](标记数组)来表示取了哪个元素,x[i]=0表示不取A[i],x[i]=1表示取A[i],与0-1背包问题的设置方法类似。

(1). DFS递归遍历

首先上场的是DFS递归遍历,利用函数的参数i表示当前进行到的位置,用has表示已经加入的元素当前的和,代码如下:

int a[] = {1,2,3,4,5};
int n = sizeof(a) / sizeof(int);
int sum = 10;

/**
 * 直接递归
 * x[]为最终解,i为考察a[i]是否加入,has表示当前的和
 */
void enum_sum_number(bool *x, int i, int has){
    if(i>=n) return;
    if(has + a[i] == sum){
        x[i] = true;
        print(a,x); // 自定义的输出函数
        x[i] = false;
    }
    x[i] = true;
    enum_sum_number(x, i + 1, has + a[i]);
    x[i] = false;
    enum_sum_number(x, i + 1, has);
}

(2). 分支限界法

然后考虑对DFS进行优化,便有了分支限界法。
考虑如何对分支进行限界,前提:数组A[0…N-1]的元素都大于0。
考察向量x[0…N-1],假定已经确定了前i个值,现在要判定第i+1个值x[i]为0还是1。
假定由x[0…i-1]确定的A[0…i-1]的和为has;A[i,i+1,…N-1]的和为residue(简记为r);

  • has + a[i] ≤ sum并且 has + r ≥ sum:x[i]可以为1;
  • has + (r - a[i]) >= sum:x[i]可以为0;

(注意,在编写代码进入分支的时候,要注意避免重复进入分支!)

代码如下:

/**
 * 分支限界
 */
void enum_sum_number2(bool *x, int i, int has, int residue){
    if(i>=n) return;
    if(has + a[i] == sum){
        x[i] = true;
        print(a,x);
        x[i] = false;
    } else if(has + residue >= sum && has + a[i] <= sum){
        // 注意此处是 else if ,因为若进入了 has + a[i] == sum分支,
        // 意味着已经选了a[i]了,所以此处不需要重复选择a[i]了
        x[i] = true;
        enum_sum_number2(x, i + 1, has + a[i], residue - a[i]);
    }
    if(has + residue - a[i] >= sum){
        x[i] = false;
        enum_sum_number2(x, i + 1, has, residue - a[i]);
    }
}

数理逻辑的重要应用:分支限界的条件

思考:

  • 分支限界的条件是充分条件吗? (不是,是必要条件)
  • 在新题目中,如何发现分支限界的条件。

(学会该方法,比此问题本身更重要)

(3). 考虑负数的情况

给出一个例子:数组 -3,-5,-2,4,2,1,3 , 给定sum = 5,
符合要求的数有
-3, -2, 4, 2, 1, 3
-3, 4, 1, 3
-5, 4, 2, 1, 3
-2, 4, 2, 1
-2, 4, 3
4, 1
2, 3

DFS因为是枚举,肯定能够得到正确解,关键是如何对负数的情况进行分支限界?

下面给出一种方法:

可对整个数组A[0…N-1]进行正负排序,使得负数都在前面,正数都在后面(只需要保证负数部分在前面即可,不需要保证负数部分内部也是有序的),使用剩余正数的和作为分支限界的约束:

  • 如果A[i]为负数:如果全部正数都算上还不够,就不能选A[i];
  • 如果递归进入了正数范围,按照数组是全正数的情况正常处理;

代码如下:

/**
 * 一种正负排序的实现,并计算negative和positive
 */
void positive_negative_sort(int a[], int n, int &negative, int &positive){
    int k = 0;
    negative = positive = 0;
    for(int i=0; i= n) return;
    if(has + a[i] == sum){
        x[i] = true;
        print(a,x);
        x[i] = false;
    } 
    if(a[i] > 0){
        // 正数的情况
        if(has + positive >= sum && has + a[i] <= sum){
            x[i] = true;
            enum_sum_number3(x, i + 1, has + a[i], negative, positive - a[i]);
            x[i] = false;
        }
        if(has + positive - a[i] >= sum){
            x[i] = false;
            enum_sum_number3(x, i + 1, has, negative, positive - a[i]);
        }
    } else {
        // 负数的情况
        if(has + x[i] + positive >= sum){
            x[i] = true;
            enum_sum_number3(x, i + 1, has + a[i], negative - a[i], positive);
            x[i] = false;
        }
        if(has + negative <= sum && has + positive >= sum){
            x[i] = false;
            enum_sum_number3(x, i + 1, has, negative - a[i], positive);
        }
    }
}

你可能感兴趣的:(算法,数据结构,c++,数据结构与算法)