2023“钉耙编程”中国大学生算法设计超级联赛（2）Klee likes making friends_klee喜欢交朋友,现在有n个人站在一个队伍里,klee可以和其中的一个人交朋友,这个人-CSDN博客

本文链接：https://blog.csdn.net/tanjunming2020/article/details/131903406

文章描述了一个编程竞赛题目，小K需要在一组人中选择朋友，使得每连续m个人中有至少两人是朋友，目标是最小化成本。通过动态规划方法，计算最后两个朋友间位置的最小代价，并利用模运算降低空间复杂度至O(m^2)。代码示例展示了如何实现这一策略，时间复杂度为O(nm)。

摘要由CSDN通过智能技术生成

2023“钉耙编程”中国大学生算法设计超级联赛（2）Klee likes making friends

题目大意

小 $K$ 喜欢交朋友。现在有 $n$ 个人站在小 $K$ 前面，小 $K$ 选择第 $i$ 个人交朋友的代价为 $a_i$ 。小 $K$ 希望在每连续 $m$ 个人中至少有两个人是她的朋友，请帮她求出能满足条件的最小花费。

$2\leq n\leq 2\times 10^4,2\leq m\leq 2\times 10^3,m\leq n$

数据保证 $\sum n\leq 5\times 10^4$

题解

设 $f_{i,j}$ 表示最后两个朋友的在 $i, j (i > j)$ 位置上的最小代价。因为每连续 $m$ 个人中至少有两个人是小 $K$ 的朋友，所以 $i - j < m$ ，且倒数第三个朋友的位置 $k$ 在区间 $[j - m, k - 1]$ 上。那么，转移式为

$f_{i,j}=a_i+\min\limits_{k=j-m}^{k-1}\{f_{j,k}\}$

这里可以求一个后缀最小值。

这样做的话，空间复杂度为 $O(n^2)$ ，会 $\text{MLE}$ ，我们考虑优化。

可以发现 $f_{ij}$ 中 $i$ 和 $j$ 的差不超过 $m$ ，所以我们可以用取模和位置差的方式优化，来降低空间复杂度。设 $f_{i,j}$ 表示倒数第一个朋友的位置模 $m$ 后的值为 $i$ ，倒数第一个朋友和倒数第二个朋友的位置的差为 $j$ 时的最小花费，这样空间复杂度就降到了 $O(m^2)$ 。

时间复杂度为 $O (nm)$ 。

code

#include<bits/stdc++.h>
using namespace std;
int t,n,m,ans,a[200005],f[2005][2005],g[2005][2005];
int main()
{
	scanf("%d",&t);
	while(t--){
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++){
			scanf("%d",&a[i]);
		}
		for(int i=0;i<=m;i++){
			for(int j=0;j<=m;j++) f[i][j]=g[i][j]=2e9;
		}
		for(int i=2;i<=m;i++){
			for(int j=i-1;j>=1;j--){
				f[i%m][i-j]=a[i]+a[j];
				g[i%m][i-j]=min(g[i%m][i-j-1],f[i%m][i-j]);
			}
		}
		for(int i=m+1;i<=n;i++){
			for(int j=i-1;j>=i-m+1;j--){
				f[i%m][i-j]=a[i]+g[j%m][j-(i-m)];
				g[i%m][i-j]=min(g[i%m][i-j-1],f[i%m][i-j]);
			}
		}
		ans=2e9;
		for(int i=n-m+1;i<=n;i++){
			for(int j=n-m+1;j<=i-1;j++){
				ans=min(ans,f[i%m][i-j]);
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}