数学考试 牛客网

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
 

题目描述

今天qwb要参加一个数学考试，这套试卷一共有n道题，每道题qwb能获得的分数为ai，qwb并不打算把这些题全做完，
他想选总共2k道题来做，并且期望他能获得的分数尽可能的大，他准备选2个不连续的长度为k的区间,
即[L,L+1,L+2,....,L+k-1]，[R,R+1,R+2,...,R+k-1]（R >= L+k）。

输入描述:

第一行一个整数T（T<=10）,代表有T组数据
接下来一行两个整数n,k,(2<=n<=200,000),(1<=k,2k <= n)
接下来一行n个整数a1,a2,...,an，（-100,000<=ai<=100,000）

输出描述:

输出一个整数，qwb能获得的最大分数

示例1

输入

2
6 3
1 1 1 1 1 1
8 2
-1 0 2 -1 -1 2 3 -1

输出

6
7

方法

前缀和预处理加动态规划

#include <iostream>
using namespace std;

int arr[200000];
long long b[200000];

void init(int n)
{
	for (int i = 0; i < n; i++)
	{
		cin >> arr[i];
	}
	b[0] = arr[0];
	for (int i = 1; i < n; i++)
	{
		b[i] = arr[i] + b[i - 1];
	}
}

long long getsum(int l, int r)
{
	return b[r] - b[l] + arr[l];
}

long long dp[200005][2];
long long process2(int n, int k, int index, int times)
{
	for (int i = n - 1 * k; i >= k; i--)
	{
		dp[i][0] = max(dp[i + 1][0], getsum(i, i + k - 1));
	}
	dp[n - 2 * k][1] = getsum(n - 2 * k, n - 1);
	for (int i = n - 2 * k - 1; i >= 0; i--)
	{
		dp[i][1] = max(dp[i + 1][1], getsum(i, i + k - 1) + dp[i + k][0]);
	}
	return dp[index][times];
}


int main()
{
	int T;
	cin >> T;
	while (T--)
	{
		int n, k;
		cin >> n >> k;
		init(n);
		if (n / k < 2)
		{
			cout << 0 << endl;
			continue;
		}
		long long num2 = process2(n, k, 0, 1);
		cout << num2 << endl;
	}
	return 0;
}