T1
DQS 捡石子
题目描述
今年开始上学的 DQS 非常喜欢石子, 她总是会收集很多不同类型的石子来卖钱, 这个世界的石子只有两种——蓝色和白色(用01 表示) 并且都是连在一起的, 不能移动, 因此 DQS 只好使用她的神力来解除石子不能移动的封印, 但是由于某些原因DQS 希望让自己消耗更多的神力, 因此她许愿黑暗之神让她可以转换连在一起的石子中的一颗。 消耗的神力计算方法为这一串石子中相邻相同的石子个数的平方和。 DQS 想知道, 如何改变其中一个石子的种类, 使得整个石子串的消耗最大(含多组数据)
输入描述
一个数 T
接下来 T 行, 每行一个长度为 N 的 01 串
输出描述 一个数 p 表示 DQS 消耗的神力
样例输入
2
000011
0101
样例输出
26
10
数据范围及提示 1<=T<=50
60%:1<=N<=1000
100%:1<=N<=100000
纯暴力,没什么技术含量,但是在这道题上跪了。一开始的思路有错误,认为一定要先找出连续次数最多的一组数,在这组数的左边或右边进行操作,还没来得及验证就急急忙忙开始打了,刚刚培养的小习惯也丢了。
正解:先将整个串扫一遍,把个连续子串的位置以及具体数目存到另一个数组中,对这个数组进行操作。
比较每个位置修改左边或修改右边取得的最大值,取所有最大值的最大值。
要格外注意一种情况:
0 0 0 1 0 0 0
此时修改中间的1得到最大值。
修改过程中注意判断边界,如果这个位置在最左边,那么只需要比较未修改的值和修改右边的值,最右边的位置同理。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=100000+10;
char a[maxn];
int t,n,sum;
struct node
{
int v;//0还是1
int w;//有多少个
}b[maxn];
int p(int a)
{
int d=1;
d=a*a;
return d;
}
int main()
{
//freopen("stone.in","r",stdin);
//freopen("stone.out","w",stdout);
scanf("%d",&t);
while(t--)
{
int ans;
sum=0,ans=0;
cin>>a;
int n=strlen(a);
memset(b,0,sizeof(b));
int k=1,maxx=0;
for(int i=0;i<n;i++)
{
if(a[i+1]==a[i])
{
b[k].w++;
}
else if(a[i+1]!=a[i])
{
b[k].v=a[i]-'0';
k++;
}
}
k-=1;
for(int i=1;i<=k;i++)
{
b[i].w+=1;
sum+=b[i].w*b[i].w;
}
ans=sum;
for(int i=1;i<=k;i++)
{
if(i==1)
ans=max(ans,sum-p(b[i+1].w)+p(b[i+1].w-1)-p(b[i].w)+p(b[i].w+1));
else if(i==k)
ans=max(ans,sum-p(b[i-1].w)+p(b[i-1].w-1)-p(b[i].w)+p(b[i].w+1));
else if(b[i].w==1)
ans=max(ans,sum-p(b[i-1].w)-p(b[i+1].w)+p(b[i+1].w+b[i-1].w+1)-1);
else
ans=max(ans,max(sum-p(b[i-1].w)+p(b[i-1].w-1)-p(b[i].w)+p(b[i].w+1),sum-p(b[i+1].w)+p(b[i+1].w-1)-p(b[i].w)+p(b[i].w+1)));
}
if(k==1) cout<<sum<<'\n';
else
cout<<ans<<'\n';
}
//fclose(stdin);
//fclose(stdout);
return 0;
}