迅雷笔试题_红黑积木求和

题目：

有红黑两种颜色的方块积木，红色代表正数A、黑色代表负数B，选出17块积木排成一排，使得任意相邻7块积木之和小于0；如何挑选才能使得17块积木之和最大，最大值是多少？

输入：

正数A、负数B；

A和B的绝对值均小于10000；

样例输入：

10  -61

样例输出：

28

 

思路：

用一个长度为7的滑动窗口即可解决问题。

实现：

#include <iostream>
#include <bits/stdc++.h>
using namespace std;

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>

int main() {

    int a,b;
    cin >> a >> b;
    int count = 0;
    int sum =0;
    int num[17];
    for (int i=0;i<7;i++){
        if(a*(i+1)+b*(6-i)<0){
            num[i]=a;
            sum = sum + num[i];
            count = count + num[i];
        }else {
            num[i]=b;
            sum = sum + num[i];
            count = count + num[i];
        }
    }


    for (int i=7;i<17;i++){
        if (count - num[i-7]+a<0){
            num[i]=a;
            count = count - num[i-7]+a;
            sum = sum + num[i];
        }else {
            num[i]=b;
            count = count - num[i-7]+b;
            sum = sum + num[i];
        }
    }
    cout << sum << endl;
}