bdfz_2024_summer/day2/U111091/teacher_sun.cpp

99 lines
2.3 KiB
C++
Raw Normal View History

2024-08-05 06:31:16 +00:00
#include<bits/stdc++.h>
#define MAXN 200005
#define P int(1e9+7)
using namespace std;
typedef long long ll;
/*
U111091 2
k==1
1
k==2
1.
2. i2i之前和i之后
2
2使
case1
case2dp之后维护前缀/max
线
*/
int T,N,K,q;
int x[MAXN], sum = 0;
int g[MAXN], pre[MAXN], suf[MAXN];
char s[MAXN];
int solve1(int q)//解决优化一段长度为q的土路的情况
{
int k = 1;//一段路的开始位置
int ans = 0;//能优化的最长的土路的长度
int ans1;
for(int i=2;i<=N;i++)
{
while(x[i]-x[k]>q)
k++;
ans1 = g[i] - g[k]; //第k个村庄到第i个村庄之间土路的长度
if(s[k]=='1')
ans1 += q - (x[i]-x[k]);
ans = max(ans, ans1);
}
//cerr<<"ans1 = "<<ans1<<endl;
return ans;
}
int solve2(int q)//计算找两段长度为q的土路最多能改变多少变成高速公路
{
memset(pre, 0, sizeof(pre));
memset(suf, 0, sizeof(suf));
int ans1;
int k = 1;//一段路的开始位置
for(int i=2;i<=N;i++){
while(x[i]-x[k]>q) ++k;
ans1 = g[i] - g[k];
if(s[k]=='1')
ans1 += q - (x[i]-x[k]);
pre[i] = max(pre[i-1], ans1);
}
k = N;
for(int i=N-1;i>=1;i--){
while(x[k]-x[i]>q)
--k;
ans1 = g[k] - g[i];
if(s[k+1]=='1')
ans1 += q - (x[k]-x[i]);
suf[i] = max(suf[i+1], ans1);
}
int ans = 0;
for(int i=2;i<=N;i++){
ans = max(ans, pre[i] + suf[i]);
}
return ans;
}
int main(){
cin>>T;
while(T--){
cin>>N>>K>>q;
for(int i=1;i<=N;i++)
cin>>x[i];
cin>>s+2;
s[1] = '0';
sum = 0;//土路总长度
for(int i=2;i<=N;i++){
g[i] = g[i-1];//前i个村庄之间土路的总长度
if(s[i]=='1')//i-1到i村庄之间的路是土路
{
g[i] += x[i] - x[i-1];
sum += x[i] - x[i-1];
}
}
int ans;
if(K==1)
ans = sum - solve1(q);
else
ans = sum - max(solve1(2*q), solve2(q));
cout<<ans<<'\n';
}
return 0;
}