【csp模拟赛1】不服来战 (challenge.cpp)
作者：互联网
【题目描述】
最近小 Z 和他的朋友都迷上了一款手机游戏：不服来战。 游戏的设定十分简单，在游戏开始时，会给出一排共 N 个灯，有的灯是开着 的有的是关着的，每个灯都有一个分数。而玩家可以进行任意次操作，每次操作 改变连续 K 盏灯的开关状态。尽管机智如小 Z 也总是没法得到最高分，没法把他 的朋友 PK 下来。于是他来向你请教，希望知道在不同情况下，最高分分别是多 少。
【输入格式】
第一行，一个正整数 T，表示测试数据组数。 对于每组测试数据： 首先是一行两个正整数 N，K，意义如题目所述。 接着是 N 个数依次表示 1~N 每盏灯的开关状态。每个数是 0 或 1，分别 表示 这盏灯初始时是关的或开的。 最后是 N 个非负整数依次表示 1~N 每盏灯亮着的得分。
【输出格式】
对每组测试数据输出一行。表示最大的得分。
【样例输入】
3 3 2 0 1 1 3 2 4 10
1 0 0 1 0 1 0 0 1 0 1
1 1 1 3 4 3 4 5 1 5 10 7
1 1 1 1 1 1 1 1 1 1
5 5 3 3 4
3 5 1 1 3
【样例输出】
7
28
33
【样例解释】
第一组数据：翻转前 2 盏灯是最优方案； 第二组数据：你可以把所有的灯点亮； 第三组数据：不做任何操作是最优方案
思路：
不从图的角度思考也可以得出正解。
考虑操作 i 和操作 i+1 同时按下，这时产生的 效果时把第 i 盏灯和第 i+K 盏灯翻转。
也就是说，我们得到了一种新的操作，这种操作是， 把任意相距为 K 的两盏灯同时翻转。
不难发现，以下两种操作可以等价于原题中的操作 “翻转任意连续 K 盏灯的状态”：
（1）同时翻转 1~K 盏灯
（2）同时翻转任意距离为 K 的两盏灯 等价性是因为，操作（2）可以由原题中的操作得来；而原题中“翻转任意连续 K 盏 灯的状态”可以通过一次（1）和若干次（2）组合而来。
下面我们只考虑新的操作（1）（2）。
首先，操作（1）是否进行可以枚举。这时，我 们只需考虑操作（2）。而操作（2）似乎把一列灯也分成了许多组。以 N=10, K=3 为例，
灯 1,4,7,10 是一组；
灯 2,5,8 是一组；
灯 3,6,9 是一组。
每次进行操作（2），等价于翻转一组内相邻的两盏灯！ 组与组之间没有影响！ 再稍作分析，不难发现，如果一组内“初始时是关的”的灯为偶数个，那么我们可以 做到只把它们全部打开，也就是说打开了这一组所有的灯；如果一组内“初始时是关的” 的灯为奇数个，这个时候我们不能把所有的灯都打开，只好作出让步，选择放弃亮度最小 的那盏灯。每一组都如此处理，最后加起来。
解法一、二的时间复杂度都是 O(N)的。
代码：
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cstring>
const int MAXN = 1000000;
using namespace std;
int T,n,k,val[MAXN],vis[MAXN],sum;
inline int read()
{
int x = 0,f = 1; char s = getchar();
while(s < '0' || s > '9') {if(s == '-') f = -1;s = getchar();}
while(s >= '0' && s <= '9') {x = x * 10 + s - '0';s = getchar();}
return x * f;
}
void clean()
{
memset(val,0,sizeof(val));
memset(vis,0,sizeof(vis));
sum=0;
}
int get1()
{
int ans1=sum;
for(int i=1;i<=k;i++)
{
int tot=0,mi=2147483647;
for(int j=i;j<=n;j+=k)
{
if(vis[j]==0)tot++;
mi=min(mi,val[j]);
}
if(tot & 1)ans1-=mi;
}
// cout<<ans1<<endl;
return ans1;
}
int get2()
{
int ans2=sum;
for(int i=1;i<=k;i++)vis[i]^=1;
for(int i=1;i<=k;i++)
{
int tot=0,mi=2147483647;
for(int j=i;j<=n;j+=k)
{
if(vis[j]==0)tot++;
mi=min(mi,val[j]);
}
if(tot & 1)ans2-=mi;
}
// cout<<ans2<<endl;
return ans2;
}
void work()
{
n=read();k=read();
for(int i=1;i<=n;i++) vis[i]=read();
for(int i=1;i<=n;i++) val[i]=read(), sum+=val[i];
int s1=get1();
int s2=get2();
int ans=max(s1,s2);
printf("%d\n",ans);
}
int main()
{
#ifdef yilnr
#else
freopen("challenge.in","r",stdin);
freopen("challenge.out","w",stdout);
#endif
T=read();
while(T--)clean(),work();
fclose(stdin);fclose(stdout);
return 0;
}
完毕：
一lin
标签：盏灯,10,一组,challenge,来战,cpp,操作,include,翻转 来源： https://www.cnblogs.com/yelir/p/11524184.html