循环不变式
何为循环不变式?
类比于数学中的数学归纳法。循环不变式分为三个部分,分别是初始化、保持、终止。
- 初始化:在没有进入循环之前,这个循环不变式依然满足不变的属性。
- 保持:在每一次循环开始之前,这个循环不变式依然满足不变的属性。
- 终止:循环可以被证明终止,终止时,这个循环不变式依然满足不变的属性,此时就是答案的解。
因此,在三个部分重复的一句话就是,这个循环不变式依然满足不变的属性。循坏不变式在任何时候都保持不变的特征。
循环不变式是证明算法正确性的一个重要的方法,也是设计迭代算法重要的一依据。
介绍循环不变式比较好的书籍应该是《编程珠玑》,有空应该再看一遍。
例子
下面的代码,设集合A为已经处理完成的数组元素,集合A‘为未处理的数组元素,我们的循环不变式的断言是:
在每一次循环开始之前,step都是处理A所需要的全部最小步骤。req变量是A’中第一个元素要满足的数量。
我们通过循环不变式的三个部分,证明这个循环不变式的正确性。
- 初始化:此时A为空,对应step为0,A‘就是全部元素,req就是让A’的第一个元素,即原数组的第一个元素让他变成平均数。
- 保持:如果A‘的第一个元素满足req,那么就把A’的第一个元素移动到A中,并且将req设置为平均数,即要求下一个元素应该是平均数。如果A‘的第一个元素大于req,因为A已经处理完成了,我们不能向A中移动了,因此我们只能向A’的下一个元素一次性的移动,满足req,步骤为1,且步骤最小,并且将req设置为平均数,即要求下一个元素应该是平均数。如果A‘的第一个元素小于req,同理,我们不能向A中索取纸牌,因此只能向A’中的元素社区纸牌,此时req应该是平均数加上需求量,步骤为1,且步骤最小。
- 终止:此时A‘为空,A为所有元素,此时step就是处理A所需要的最小步数,即答案的解。
#include <bits/stdc++.h>
using namespace std;
int main()
{
int n;
cin >> n;
int arr[10001];
int sum = 0;
for (int i = 0; i < n; i++)
{
cin >> arr[i];
sum += arr[i];
}
int tag = sum / n;
int step = 0;
int req = tag;
for (int i = 0; i < n - 1; i++)
{
if (arr[i] == req)
{
req = tag;
}
else if (arr[i] > req)
{
arr[i + 1] += arr[i] - req;
req = tag;
step++;
}
else
{
req = tag + req - arr[i];
step++;
}
}
cout << step;
return 0;
}
总结
循环不变式的魅力还不止如此,如果我们迭代设计算法的时候能想着循环不变式,那么将大大降低我们出错的几率。