HDU

来源:互联网 发布:狮王酵素牙膏知乎 编辑:程序博客网 时间:2024/06/12 19:27

题目点此跳转

思路

 题目意思是给你一全环形的数组(头尾相接), 求所有长度不大于k的区间中 元素和 最大 的的区间 及 最大的元素和。

 区间的和可以使用前缀和相减求出,设sum[i]为从0到i的前缀和, 区间[i, j]的和即为sum[j] - sum[i-1]; 那么对于每一个区间尾j,我们只要求出它前面i个内最小的sum[i]即可,显然单调队列是可以解决的。
 每次区间尾j右移时,将j所在元素入队(注意删除队列中比它大的), 然后将不在此区间内的队首元素出队,剩下的队首元素即为此区间内的最小值。
 注意此题的数组是环形的,多解时输出的顺序也有要求,所以区间尾j的初始值要处理好。

代码

#include <algorithm>#include <iostream>#include <sstream>#include <utility>#include <string>#include <vector>#include <queue>#include <map>#include <set>#include <cstring>#include <cstdio>#include <cmath>#define met(a,b) memset(a, b, sizeof(a));#define IN freopen("in.txt", "r", stdin);#define OT freopen("ot.txt", "w", stdout);using namespace std;typedef long long LL;typedef pair<int, int> PII;const int maxn = 1e6 + 100;const LL INF = 0x7fffffff;const int dir[5][2] = {0,0,-1,0,1,0,0,-1,0,1};const int MOD = 1e9 + 7;const double eps = 1e-6;int n, k, a[maxn], sum[maxn], ans, L, R;int q[maxn], p[maxn], ft, rr;int main() {    #ifdef _LOCAL    IN; //OT;    #endif // _LOCAL    int t; cin >> t;    while(t--) {        scanf("%d%d", &n, &k);        for(int i = 1; i <= n; ++i) scanf("%d", &a[i]);        for(int i = 1; i < k; ++i) a[n+i] = a[i];        sum[0] = a[n];        for(int i = 1; i < n+k; ++i) sum[i] = sum[i-1] + a[i];        ft = rr = 0; ans = -INF;        q[rr++] = sum[0]; p[rr-1] = 0;        for(int i = 1; i <= k-1; ++i) {            while(rr > ft && q[rr-1] >= sum[i]) --rr;            q[rr++] = sum[i], p[rr-1] = i;        }        if(ans < sum[k]-q[ft]) { ans = sum[k]-q[ft]; L = p[ft]%n+1; R =  (k-1)%n+1; }        for(int i = k+1; i < n+k; ++i) {            while(rr > ft && q[rr-1] >= sum[i-1]) --rr;            q[rr++] = sum[i-1], p[rr-1] = i-1;            if(p[ft] < i-k) ++ft;            if(ans < sum[i] - q[ft]) { ans = sum[i]-q[ft]; L = p[ft]%n+1; R = (i-1)%n+1; }            else if(ans == sum[i] - q[ft]) {                if(L > p[ft]%n + 1) L = p[ft]%n+1, R = (i-1)%n+1;                else if( L == p[ft]%n + 1 && R-L > (i-1)%n-p[ft]%n) L = p[ft]%n+1, R = (i-1)%n+1;            }        }        printf("%d %d %d\n", ans, L, R);    }    return 0;}
原创粉丝点击