【BZOJ3827】[Poi2014]Around the world【尺取法】【并查集】

【题目链接】

对于每个询问，设油量为d。

先预处理出每个点走一次最多走到哪，这个用尺取法可以O(n)。

然后得到一颗树，算一下每个点的深度。

枚举起点，在树上一直向上爬，直到距离超过n，爬的过程同时用并查集合并。

傻逼题卡内存。

/* Pigonometry */
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn = 1000005, inf = 0x7FFFFFFF;

int n, m, dis[maxn], next[maxn << 1], depth[maxn << 1];

inline int iread() {
int f = 1, x = 0; char ch = getchar();
for(; ch < '0' || ch > '9'; ch = getchar()) f = ch == '-' ? -1 : 1;
for(; ch >= '0' && ch <= '9'; ch = getchar()) x = x * 10 + ch - '0';
return f * x;
}

inline int solve(int x) {
int fa = x;
for(; fa < x + n; fa = next[fa]);
for(int i = next[x], j = x; i != fa; j = i, i = next[i]) next[j] = fa;
return depth[x] - depth[fa];
}

int main() {
n = iread(); m = n << 1;
int k = iread(), lwb = 0;
for(int i = 1; i <= n; i++) {
dis[i] = iread();
lwb = max(lwb, dis[i]);
}
while(k--) {
int d = iread();
if(d < lwb) {
printf("NIE\n");
continue;
}
for(int i = 1, j = 1, sum = 0; i <= m; i++) {
while(j < m)
if(sum + dis[j > n ? j - n : j] <= d) {
sum += dis[j > n ? j - n : j];
j++;
}
else break;
sum -= dis[i > n ? i - n : i];
next[i] = j;
}
depth[m] = 0;
for(int i = m; i >= 1; i--) depth[i] = depth[next[i]] + 1;
int ans = inf;
for(int i = 1; i <= n; i++) ans = min(ans, solve(i));
printf("%d\n", ans);
}
return 0;
}