1121 - Reverse the lights 思维题

​​http://www.ifrog.cc/acm/problem/1121​​

我看到这些翻转的题就怕，可能要练下这些专题。

我最怕这类题了。

一开始想了下dp， dp[i][0 / 1]表示完成了前i位，第i位不按 / 按，的状态，然后发现转移不了。无果。好像是按下这一位，然后后面的k个又会变，表示不了。

然后来了一个贪心，对于第1个，在[1, k + 1]之间，肯定要按下一个了，那么按哪一个呢？我自己写了一个函数来判断按下这一个位的价值，就是，[pos - k, pos + k]数中所有的和，然后按下了这位，就跳去下一个没亮的区间按，然后这个有反例。

7 2
1 0 500 1 100000 0 100000

然后就觉得自己做不出了，看到别人那里说，按下的灯应该间隔2 * k。也就是不应该有重叠，枚举k + 1个区间就好。

想想还真有道理。无奈想不到。

#include <bits/stdc++.h>
#define IOS ios::sync_with_stdio(false)
using namespace std;
#define inf (0x3f3f3f3f)
typedef long long int LL;
const int maxn = 1e6 + 20;
int a[maxn];
LL sum[maxn];
void work() {
    int n, k;
    scanf("%d%d", &n, &k);
    for (int i = 1; i <= n; ++i) scanf("%d", a + i);
    for (int i = 1; i <= n; ++i) {
        sum[i] = sum[i - 1] + a[i];
    }
    LL ans = 1e18;
    for (int i = 1; i <= min(n, k + 1); ++i) { 
        LL t = 0;
        for (int j = i; j <= n; j = j + 2 * k + 1) {
            t += a[j];
            if (j + 2 * k + 1 > n && j + k < n) {
                t = 1e18;
                break;
            }
        }
        ans = min(ans, t);
    }
    cout << ans << endl;
}

int main() {
#ifdef local
    freopen("data.txt", "r", stdin);
//    freopen("data.txt", "w", stdout);
#endif
    work();
    return 0;
}

View Code

7 9
1 1 500 1 100000 1 100000