Codeforces Round #450 (Div. 2) E. Maximum Questions（线段树+DP）

题目链接：http://codeforces.com/contest/900/problem/E

dp[i]表示从i开始答案最大为多少，处理出每个位置最多向后延伸多少，然后从n到1开始dp，用线段树维护最大值就好了。

代码：

#include<bits/stdc++.h>
using namespace std;
const int MAXN=1e5+5;
char s[MAXN];
struct node
{
	int ans,cost;
	node(int _ans=0,int _cost=0):ans(_ans),cost(_cost){}
	void add(const node o)
	{
		ans+=o.ans;
		cost+=o.cost;
	}
	bool operator < (const node &o)const
	{
		if(ans==o.ans)
			return cost>o.cost;
		return ans<o.ans;
	}
};
struct seg
{
	#define lson l,mid,rt<<1
	#define rson mid+1,r,rt<<1|1
	node mx[MAXN<<2];
	void push_up(int rt)
	{
		mx[rt]=max(mx[rt<<1],mx[rt<<1|1]);
	}
	void update(int pos,node val,int l,int r,int rt)
	{
		if(l==r)
		{
			mx[rt]=max(mx[rt],val);
			return ;
		}
		int mid=(l+r)>>1;
		if(mid>=pos)
		{
			update(pos,val,lson);
		}
		if(mid<pos)
		{
			update(pos,val,rson);
		}
		push_up(rt);
	}	
	node query(int L,int R,int l,int r,int rt)
	{
		if(L<=l&&r<=R)
		{
			return mx[rt];
		}
		int mid=(l+r)>>1;
		node ret;
		if(L<=mid)
			ret=max(ret,query(L,R,lson));
		if(mid<R)
			ret=max(ret,query(L,R,rson));
		return ret;
	}
}se;
int num[MAXN],sa[MAXN];
int main()
{
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	int n,m;
	scanf("%d",&n);
	scanf("%s",s+1);
	scanf("%d",&m);
	for(int i=1;i<=n;i++)
	{
		if(s[i]=='?')
			num[i]=1;
		num[i]+=num[i-1];
	}
	for(int i=n;i>=1;i--)
	{
		if(s[i]=='?'||s[i]=='a')
		{
			if(s[i+1]=='b'||s[i+1]=='?')
				sa[i]=sa[i+2]+2;
			else
				sa[i]=1;
		}
		else
		{
			sa[i]=0;
		}
		node now=se.query(i,n,1,n,1);
		if(sa[i]>=m)
		{
			node tmp=se.query(i+m,n,1,n,1);
			tmp.add(node(1,num[i+m-1]-num[i-1]));
			now=max(now,tmp);
			se.update(i,now,1,n,1);
		}
	}
	node ans=se.query(1,n,1,n,1);
	printf("%d\n",ans.cost);
	return 0;
}