HDU4339. Query(树状数组+二分）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4339
题意：对两个字符串，1操作为改变一个串的一个字符，2为从一个位置开始计算s1与s2的最长相同串
解题思路：
树状数组，将s1与s2相同的更新为1，不相同的更新为0，然后利用二分法去区间求和，找到区间和=区间长度的位置，即可。注意：s1与s2的长度可能不一样

#include<iostream>
#include<cstdio>
#include<string>
#include<string.h>
using namespace std;
int n,q;
int len1,len2,len;
int c[1100000],a[1100000];
char s1[1100000],s2[1100000];
int min(int a,int b){
	return a>b?b:a;
}
int lowbit(int x){
	return x&(-x);
}
void update(int i,int k){
	while(i<=n){
		c[i]+=k;
		i+=lowbit(i);
	}
}
int get_sum(int i){
	int res=0;
	while(i>0){
		res+=c[i];
		i-=lowbit(i);
	}
	return res;
}
int binarysearch(int l)
{
	int bv=get_sum(l-1);
	int begin=l;
	int r=n,mid;
	while(l<=r){
		mid=(l+r)/2;
		int ret=get_sum(mid)-bv;
		if(ret<(mid-begin+1))
			r=mid-1;
		else
			l=mid+1;
	}
	return (r-begin+1);
}
int main()
{
	int t;
	int cnt=1;
	cin>>t;
	while(t--){
		memset(a,0,sizeof(a));
		memset(c,0,sizeof(c));
		scanf("%s%s",s1+1,s2+1);
		len1=strlen(s1+1);
		len2=strlen(s2+1);
		len=min(len1,len2);
		n=len;
		for(int i=1;i<=len;i++){
			a[i]=(s1[i]==s2[i]);
			update(i,a[i]);
		}
		scanf("%d",&q);
		printf("Case %d:\n",cnt++);
		int aa,bb,cc;
		char d;
		for(int i=0;i<q;i++){
			scanf("%d",&aa);
			if(aa==2){
				scanf("%d",&bb);
				if(bb>len){
					printf("0\n");
					continue;
				}
				printf("%d\n",binarysearch(bb+1));
			}
			if(aa==1){
				scanf("%d %d %c",&bb,&cc,&d);
				cc++;
				if(bb==1){
					if(cc>len)
						continue;
					s1[cc]=d;
					int temp=(s1[cc]==s2[cc]);
					if(temp!=a[cc]){
						a[cc]=temp;
						update(cc,(temp==1?1:-1));
					}
				}
				if(bb==2){
					if(cc>len)
						continue;
					s2[cc]=d;
					int temp=(s1[cc]==s2[cc]);
					if(temp!=a[cc]){
						a[cc]=temp;
						update(cc,(temp==1?1:-1));
					}
				}
			}
		}
	}
	return 0;
}