hdu 4339 Query ---线段树

给两个字符串，

1操作修改其中一个串的一个字符的值

2操作询问从 该字符串开始往后的 两个字符串 字符相同的长度

标记 字符相同为1，不同为0，每次修改相当于修改0,1值，

zhi,标记每段的左值 ，为0，或1

la,标记每段从左开始连续1的长度

flag,标记叶子节点的1,0值//没啥用，可删掉。。

len ,标记段长

然后就是更新 ，询问操作了。。

#include<stdio.h>  
#include<string.h>  
#include<math.h>  
#include<stdlib.h>  
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
#define ll __int64
#define inf 0x3f3f3f3f
#define N 1001000

char a[N],b[N];
int al,bl;
struct node
{
	int l,r,la,flag,zhi,len;
}tree[4*N];

void build(int id,int x,int y)
{
	tree[id].l=x;
	tree[id].r=y;
	if(x<=al && x<=bl && a[x]==b[x])
		tree[id].zhi=1;
	else tree[id].zhi=0;
	tree[id].len=y-x+1;
	if(x==y){
		if(x<=al && x<=bl && a[x]==b[x])
			tree[id].flag=tree[id].la=tree[id].zhi=1;
		else tree[id].flag=tree[id].la=tree[id].zhi=0;
		return ;
	}
	int mid=(x+y)>>1;
	build(id<<1,x,mid);
	build((id<<1)+1,mid+1,y);
	if(tree[id<<1].len==tree[id<<1].la && tree[(id<<1)+1].zhi==1)
		tree[id].la=tree[id<<1].len+tree[(id<<1)+1].la;
	else tree[id].la=tree[id<<1].la;
}
void update(int id,int x,int y)
{
	if(tree[id].l==tree[id].r && tree[id].l==x)
	{
		tree[id].la=tree[id].flag=tree[id].zhi=y;
		return ;
	}
	if(tree[id].l==x)
		tree[id].flag=tree[id].zhi=y; //更新每段的左值。。（尼玛又忘了，害我调试半天

	int mid=(tree[id].l+tree[id].r)>>1;
	if(x<=mid) update(id<<1,x,y);
	else update((id<<1)+1,x,y);
	if(tree[id<<1].len==tree[id<<1].la && tree[(id<<1)+1].zhi==1)
		tree[id].la=tree[id<<1].len+tree[(id<<1)+1].la;
	else tree[id].la=tree[id<<1].la;
}
int query(int id,int x)
{
	if(tree[id].l==tree[id].r && x==1)
		return tree[id].la;
	if(tree[id].la>=x)
		return tree[id].la-x+1;
	int mid=(tree[id].l+tree[id].r)>>1;

	int t=mid-tree[id].l+1;

	if(x<=t)
	{
		int ans=query(id<<1,x);
		if(ans==tree[id<<1].len-x+1) //如果和右段相连，就加上右段的la值。。
			return ans+tree[(id<<1)+1].la;
		else
			return ans;
	}
	else 
	{
		return query((id<<1)+1,x-t);
	}
}
int main()
{
	int i,j,k,c,d,e;
	char ch,tc;
	int t;
	cin>>t;
	int ans=0;
	while(t--)
	{
		ans++;
		scanf("%s%s",a+1,b+1);
		al=strlen(a+1);
		bl=strlen(b+1);
		int end=max(al,bl);
		build(1,1,end);
		scanf("%d",&k);
		printf("Case %d:\n",ans);
		while(k--)
		{
			scanf("%d",&c);
			if(c==1){
				scanf("%d%d%c%c",&d,&e,&ch,&ch);
				e++;
				if(d==1) a[e]=ch;
				else b[e]=ch;
				if(a[e]==b[e])//这里可以多加些if else优化一下，懒得写了。。
					update(1,e,1);
				else
					update(1,e,0);
			}
			else if(c==2){
				scanf("%d",&d);
				d++;
				int ans=query(1,d);
				printf("%d\n",ans);
			}
		}
	}
}