C++循环边界的计算

假设存在一个数组A

[1, 1, 1, 1, 1],

现要计算将数组的每一个元素与其前后元素相加构成新数组，可以使用如下if语句，

for (int i = 0; i < 5; i++){
    if (i == 0){
        B[i] = A[4] + A[0] + A[1];
    }
    else if (i == 4){
        B[i] = A[3] + A[4] + A[0];
    }
    else{
        B[i] = A[i-1] + A[i] + A[i+1];
    }
}

也可以采用三元运算符简化写法，

for (int i = 0; i < 5; i++){
    B[i] = A[i - 1 + (i - 1 < 0 ? 5 : 0)] + A[i] + A[i + 1 + (i + 1 > 4 ? -5 : 0)];
}

如若对前后元素进行更复杂的操作，如新数组的对应的元素的值为前一个元素的两倍加上原元素的三倍加上后一个元素的四倍，则可使用如下方法，

int t[3] = {2, 3, 4};
for (int i = 0; i < 5; i++) {
	int num = 0;
	for (int j = 0; j < 3; j++) {
		num += A[i + j - 1 + (i + j - 1 < 0 ? 5 : 0) + (i + j - 1 > 4 ? -5 : 0)] * t[j];
	}
	B[i] = num;
}

如果是对多维数组进行复杂操作，使用三元运算符可以大幅度简化代码。