HDU 4339 Query (线段树)

题意： 有两个字符串，给出 Q 个询问，每个询问有两种体式格式：

1 p i c 把第 p 个字符串的第i 个字符换成 字符 c，

2 i     从第位i 开始，两个字符串连续相同的子串的最大长度为多少。

思路：普通的查找肯定超时，所以用线段树 复杂度大概为O（n + Qlog(n)） 每个结点保存该区间内第一个不匹配的下标，query时，只需找到与key（要查找的点） 最近的一个下标 相减就得

//7156MS    26832K
#include <stdio.h>
#include <string.h>
#define L(x) (x<<1)
#define R(x) (x<<1|1)
#define M 1000050
int inf,a,b;
struct data
{
    int l,r,lab; //lab表示在当前区间内第一个不匹配的下标
}node[M*3];

char str[2][M];

int min (int a,int b)
{
    return a > b ? b : a;
}
void BuildTree (int left,int right,int u)
{
    node[u].l = left;
    node[u].r = right;
    node[u].lab = inf;
    if (left == right)
        return ;
    int mid = (left + right)>>1;
    BuildTree(left,mid,L(u));
    BuildTree(mid+1,right,R(u));
}

void update (int key,int lab,int u)
{
    if (node[u].l == node[u].r)
    {
        node[u].lab = lab;
        return ;
    }
    int mid = (node[u].l + node[u].r)>>1;
    if (key <= mid)
        update (key,lab,L(u));
    else
        update (key,lab,R(u));
    node[u].lab = min (node[L(u)].lab,node[R(u)].lab);
}

int query (int key,int u)
{
    if (key <= node[u].l)    //只需要找>= key 的范围
        return node[u].lab;
    if (key <= node[L(u)].r)
        a = query (key,L(u));    //a 原初始化为inf
    b = query (key,R(u));          //右边的肯定满足
    return min (a,b);
}
int main ()
{
    int T,Q,k,i,op,s,count = 0;
    char c;
    scanf ("%d",&T);
    while (T --)
    {
        scanf ("%s%s",str[0],str[1]);
        int len = min (strlen(str[0]),strlen(str[1]));
        inf = len + 1;
        BuildTree(1,len,1);
        for (i = 0;i < len;i ++)
            if (str[0][i] != str[1][i])
                update (i+1,i+1,1);
            else
                update (i+1,inf,1);
        scanf ("%d",&Q);
        printf ("Case %d:\n",++count);
        while (Q --)
        {
            scanf ("%d",&op);
            if (op == 2)
            {
                scanf ("%d",&k);
                a = inf;
                int ans = query(k+1,1) - (k+1);
                printf ("%d\n",ans);
            }
            else
            {
                scanf ("%d %d %c",&s,&k,&c);
                --s;                                            //只有两种情况需要更新
                if (str[s][k] == str[s^1][k]&&str[s^1][k] != c)//原来两字符相等 改后不等了
                    update(k+1,k+1,1);
                if (str[s][k] != str[s^1][k]&&str[s^1][k] == c)//原两字符不等 改后相等了
                    update (k+1,inf,1);
                str[s][k] = c;
            }
        }
    }
    return 0;
}