题意:有n个房间,需要把n个房间全部连上网,可以使用i的代价直接将第i个房间连上网。有一些房间可以放路由器,代价也是他的标号i,路由器可以使得i-k,到i+k之间所有房间通上网,求最小代价使得全部房间通网。
分析:从1到n+k枚举路由器区间右端点,如果i-k是可以放路由器的房间,可以覆盖到i-2*k到i,那么dp[i]=min(dp[i],dp[j]+i),dp[j]应该为dp[i-2*k-1]到dp[i]之间的最小值。可以用数据结构维护这个最小值,比如单调队列。
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
#include <bits/stdc++.h> #define mp make_pair #define debug(x) cout << #x << ": " << x << endl #define pb push_back typedef long long LL; const int maxn = 4e5 + 10; const int inf = 0x3f3f3f3f; const LL INF = 0x3f3f3f3f3f3f3f3f; using namespace std; int n, k; LL dp[maxn], ans = INF; char str[maxn]; deque<int> q; int main() { scanf("%d%d", &n, &k); scanf("%s", str + 1); q.pb(0); for (int i = 1; i <= n + k; ++i) { dp[i] = dp[i - 1] + i; if (i - k >= 1 && str[i - k] == '1') { while (!q.empty() && q.front() < i - 2 * k - 1) q.pop_front(); dp[i] = min(dp[i], dp[q.front()] + i - k); } while (!q.empty() && dp[q.back()] >= dp[i]) q.pop_back(); q.pb(i); } for (int i = n; i <= n + k; ++i) ans = min(ans, dp[i]); printf("%lld ", ans); }