Problem
给定一个长度为n的01串,选一个长度至少为L的连续子串,使得子串中数字的平均值最 大。如果有多解,子串长度应尽量小;如果仍有多解,起点编号尽量小。序列中的字符编号 为1~n,因此[1,n]就是完整的字符串。1≤n≤100000,1≤L≤1000。
例如,对于如下长度为17的序列00101011011011010,如果L=7,最大平均值为6/8(子 序列为[7,14],其长度为8);如果L=5,子序列[7,11]的平均值最大,为4/5。
Input
Your program is to read from standard input. The input consists of T test cases. The number of test cases T is given in the first line of the input. Each test case starts with a line containing two integers n (1 ≤ n ≤ 100, 000) and L (1 ≤ L ≤ 1, 000) which are the length of a binary sequence and a length lower bound, respectively. In the next line, a string, binary sequence, of length n is given.
Output
Your program is to write to standard output. Print the starting and ending index of the subsequence.
Sample Input
1 2 3 4 5 2 17 5 00101011011011010 20 4 11100111100111110000
Sample Output
Solution
说来惭愧,这道题目我WA了快一面了。一开始看错题目,提交了好多次都错了,自己还没反应过来,一个劲傻傻地改。后来发现的时候,已经没有做下去的兴致了。
不过最终还是把题目完成了的。总的来说,这道题目还是让我有些收获的。题目的解题思路是通过将数列转化成坐标轴上的图像,平均值这时候也就变成了斜率了。然后一个个点去维护下凸函数的单调数列,找到最优解。
一开始我的cntAverage()这个函数只有三个参数,使用的是除法,提交之后花了0.1s,自己感觉慢了。于是把除法改成了现在的乘法,时间缩短一半,变成了0.05s。可见乘法和除法在计算的效率上还是相差很多的。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 #include <iostream> #include <string> using namespace std;const int maxn = 100005 ;int n, L, start, ending,temp;double maxd;int DNA[maxn],cav[maxn];string str; inline int cntAverage (int L, int r, int LL, int rr, int DNA[]) { return (DNA[r]-DNA[L])*(rr-LL) - (DNA[rr]-DNA[LL])*(r-L); } inline void changePoint (int pits,int bump) { double temp; temp = cntAverage (pits, bump, start-1 , ending, DNA); if (temp < 0 ) return ; if (temp || (bump-pits) < (ending-start+1 )){ maxd = temp; start = pits + 1 ; ending = bump; } } int main () { ios::sync_with_stdio (false ); cin.tie (0 ); int cas; cin >> cas; while (cas--){ int find = 0 , cor = -1 ; cin >> n >> L; cin.get (); getline (cin, str); for (int i = 1 ; i <= n; ++i) DNA[i] = DNA[i-1 ] + (str[i-1 ] == '1' ); maxd = DNA[L] / L, start = 1 , ending = L; for (int i = L; i <= n; ++i){ temp = i - L; while (find < cor && cntAverage (cav[cor], temp, cav[cor - 1 ], cav[cor], DNA) <= 0 ) --cor; cav[++cor] = temp; while (find < cor && cntAverage (cav[find], i, cav[find + 1 ], i, DNA) <= 0 ) ++find; changePoint (cav[find],i); } cout << start << ' ' << ending << '\n' ; } return 0 ; }