codeforces1353E K-periodic Garland

https://codeforces.com/problemset/problem/1353/E

首先下标对k取模，模数一样的吧i/k放进vector数组a[i%k]中

我们最后的答案肯定是只选择某一个i(0<=i<k)这个相位，然后取中间的一段，使得答案最小

由于存入的数字是i/k，那么最后的状态需要间隔为k，意思就是最后取的这段坐标的i/k是连续的

我们现在开始枚举一个相位 d，然后他中间为点亮的灯的坐标所在的块数为b[1],b[2],b[len]

那么，1的总数为sum,如果我们选择以b[i]结尾，能得到使答案最小的式子为

ans=min(ans,sum-(i-j+1)+(b[i]-b[j]+1)-(i-j+1))

sum-(i-j+1)表示除了相位为i的坐标中b[j]-b[i]这(i-j+1)个数字，全部要删除，t=(b[i]-b[j]+1)-(i-j+1)表示这一段实际需要(b[i]-b[j]+1)个连续的亮灯，但是实际上只有(i-j+1)个亮着，所以我们需要把剩余的t个没亮着的灯给点亮

那么就是tmp=min(b[j]+2*j)+b[i]+2*i-1，ans=min(ans,tmp+sum)；

这个tmp的表达式就是一个单调队列(其实因为无距离限制，只需要取最小值)的dp了

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxl=1e6+10;

int n,m,ans,sum,cas,k;
int b[maxl];
vector <int> a[maxl];
char s[maxl];
bool in[maxl]; 


inline void prework()
{
	ans=0;sum=0;
	scanf("%d%d",&n,&k);
	scanf("%s",s+1);
	for(int i=0;i<k;i++)
		a[i].clear();
	for(int i=1;i<=n;i++)
	if(s[i]=='1')
	{
		a[i%k].push_back(i/k);
		sum++;
	}
} 

inline void mainwork()
{
	if(sum==0)
	{
		ans=0; 
		return;
	}
	ans=sum-1;int tmp,len,l,r,d,mi;
	for(int yu=0;yu<k;yu++)
	{
		len=0;
		for(int d : a[yu])
			b[++len]=d;
		if(!len) continue;
		mi=-b[1]+2*1;
		for(int i=1;i<=len;i++)
		{
			mi=min(mi,-b[i]+2*i);
			tmp=mi+b[i]-2*i-1;
			ans=min(ans,sum+tmp);
		}
	}
	
}

inline void print()
{
	printf("%d\n",ans);
}

int main()
{
	int t=1;
	scanf("%d",&t);
	for(cas=1;cas<=t;cas++)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}