zoj Candies 贪心

根据题目可知3K位置的数量是可以直接确定的，并且n-2-3*k的也可以确定。所以只要这两部分不重合，答案就是确定的，或者有额外的位置是原来就已知，也是确定的。答案不确定的情况下，容易知道，有%3==1位置是同时取到最大值的，mod3==2也是同时取到最大值的。所以可以分别求出这两部分的最大值，然后O(1)回答。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=1e5+9,inf=1e9;
int a[maxn],sum[maxn];
int max1[maxn],max2[maxn];
int main()
{
//    freopen("in.txt","r",stdin);
int n;
while(scanf("%d",&n)!=EOF)
{
a[n+1]=a[0]=0;
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<=n;i++)
scanf("%d",&sum[i]);

for(int i=3;i<=n;i+=3)
a[i]=sum[i-1]-(sum[i-2]-a[i-3]);

for(int i=n-2;i>=1;i-=3)
a[i]=sum[i+1]-(sum[i+2]-a[i+3]);

for(int i=1;i+2<=n;i++)
{
int tmp=0,txt,ss=0;
for(int j=0;j<=2;j++)
if(a[i+j]==-1) tmp++,txt=i+j;
else ss+=a[i+j];
if(tmp==1)
a[txt]=sum[i+1]-ss;
}

for(int i=n-2;i>=1;i--)
{
int tmp=0,txt,ss=0;
for(int j=0;j<=2;j++)
if(a[i+j]==-1) tmp++,txt=i+j;
else ss+=a[i+j];
if(tmp==1)
a[txt]=sum[i+1]-ss;
}

bool flag=true;
for(int i=1;i<=n;i++)
if(a[i]==-1) flag=false;

if(flag)
{
int m;
scanf("%d",&m);
for(int i=1,tmp;i<=m;i++)
{
scanf("%d",&tmp);
printf("%d\n",a[tmp+1]);
}
}
else
{
for(int i=1;i<=n;i++)
if(a[i]!=-1)
max1[i]=max2[i]=a[i];
else
max1[i]=max2[i]=-inf;

int min1=inf,min2=inf;
max1[1]=100000;
for(int i=1;i+2<=n;i++)
{
int tmp=0,txt,ss=0;
for(int j=0;j<=2;j++)
if(max1[i+j]==-inf) tmp++,txt=i+j;
else ss+=max1[i+j];
if(tmp==1)
{
max1[txt]=sum[i+1]-ss;
min1=min(min1,max1[txt]);
}
}

max2[2]=100000;
for(int i=1;i+2<=n;i++)
{
int tmp=0,txt,ss=0;
for(int j=0;j<=2;j++)
if(max2[i+j]==-inf) tmp++,txt=i+j;
else ss+=max2[i+j];
if(tmp==1)
{
max2[txt]=sum[i+1]-ss;
min2=min(min2,max2[txt]);
}
}

int m;
scanf("%d",&m);
for(int i=1,tmp;i<=m;i++)
{
scanf("%d",&tmp);
tmp++;
if(tmp%3==0)
printf("%d\n",a[tmp]);
else if(tmp%3==1)
printf("%d\n",max1[tmp]+min1);
else
printf("%d\n",max2[tmp]+min2);
}
}
}
return 0;
}