题意:给定长度为n的01串,选一个长度至少为L的连续子串,使得子串中数字的平均值最大。
思路:
那么怎么维护下凸曲线呢?
每新加入一个点,检查它的加入是否使得原来的某些下凸点变为上凸,若是,删除这个下凸点,继续检查,直到曲线下凸。
上凸和下凸怎么计算?
假设起点候选表中最后两个点依次是 i , j ,新加入的点是 k ,只需要检查 i——k 的斜率是否小于 j——k的斜率,如果是,那么 j 就是一个上凸点,删除 j,继续向前检查。
斜率怎么算?
i——k的斜率就是(S[k] - S[i])/(k - i)。而平均值是(S[k] - S[i - 1])/(k - i + 1)。下标差1,用斜率算,最后结果+1就行了。
#include <cstdio>
const int N = 100000+5;
int sum[N]; // 前缀和
int p[N];//候选起点队列
char s[N];
// 比较[x1...x2] 和 [x3...x4]区间的斜率
int cmp(int x1, int x2, int x3, int x4){
return (sum[x2]-sum[x1])*(x4-x3) - (sum[x4]-sum[x3])*(x2-x1);
}
int main()
{
int T,n,L;
//freopen("in.txt","r",stdin);
scanf("%d",&T);
while( T-- ){
scanf("%d%d%s",&n,&L,s+1);
sum[0] = 0;
for( int i=1; i<=n; ++i )
sum[i] = sum[i-1] +s[i] - '0';
int i = 0;
int j = 0; // 栈
int ansL = 0, ansR = L;
for(int t = L; t <= n; ++t){ // 枚举区间右端点
// 把前面变成上凸的点去除
while(j-i > 1 &&cmp(p[j-2], t-L, p[j-1], t-L)>=0 ) --j;
p[j++] = t-L; // 加入一个新的候选起点
while(j-i > 1 &&cmp(p[i], t, p[i+1] ,t)<=0) ++i; //从头开始,找切点(最优),切点前面的不可能最优了,++i将其删除。
// 更新答案
int c = cmp(p[i], t, ansL, ansR);
if( c > 0 ||(c == 0 && t - p[i] < ansR - ansL) ){
ansL = p[i]; ansR = t;
}
}
printf("%d %d\n",ansL+1,ansR);
}
//fclose(stdin);
return 0;
}