hdu - 4339 - Query - 树状数组||线段树

用树状数组和二分寻找连续的1.http://acm.hdu.edu.cn/showproblem.php?pid=4339

做题过程:

开始做的时候发现自己以前做过。索性做吧。神奇的是我发现我以前用树状数组和线段是各A了一遍。真是勤奋的孩子。

我对这题的二分很是反感，记得以前做的时候就二分了很久来着，而且只是二分不会。。。。。。

对于线段树的作法我已经淡忘了，只记得树状数组的。明天来把线段树的再一写。

今天我才意识到，单点更新这东西是更新到底层，复杂度很高的，没有什么优越性。

/*
Pro: 0

Sol:

date:
*/
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <queue>
#include <set>
#include <vector>
using namespace std;
char str[2][1000009],d[10];
int Q,t,op,a,b,len,c[1000009];
void modify(int pos, int val){
while(pos <= len){
c[pos] += val;
pos += (pos & -pos);
}
}
int getsum(int pos){
int sum = 0;
while(pos){
sum += c[pos];
pos -= (pos & -pos);
}return sum;
}
int bin(int low, int high){
int mid; int sum = getsum(low - 1),index = low;
int ans = -1;
while(low <= high){
mid = (low + high) >> 1;
if(getsum(mid) - sum == mid - index + 1){
low = mid + 1;
ans = max(ans,mid - index + 1);
}
else high = mid - 1;
}
return ans;
}
int main(){
scanf("%d",&t);
for(int ca = 1; ca <= t; ca ++){
printf("Case %d:\n",ca);
memset(str[0],0,strlen(str[0]));
memset(str[1],0,strlen(str[1]));
scanf("%s%s",str[0],str[1]);
int len1 = strlen(str[0]);
int len2 = strlen(str[1]);
len = min(len1,len2);   //
memset(c,0,sizeof(c));
for(int i = 0; i < len; i ++){
if(str[0][i] == str[1][i])
modify(i + 1, 1);
}
scanf("%d",&Q);
for(int i = 0; i < Q; i ++){
scanf("%d",&op);
if(op == 2){
scanf("%d",&a);
int tmp = bin(a + 1,len);
if(tmp == -1)
puts("0");
else
printf("%d\n",tmp);//a + 1后面有多少个连续的1
}else{
scanf("%d%d%s",&a,&b,d);
a --;
int fk = (str[0][b] == str[1][b]);
str[a][b] = d[0];
int fg = (str[a][b] == str[!a][b]);

if(fk == 1 && fg == 0){//改前一样，改后不一样
modify(b + 1, -1);
}else if(fk == 0 && fg == 1){//改前不一样，改后一样
modify(b + 1 ,1);
}
}
}
}
return 0;
}

贴了以前的代码，因为以前的注释多。仍是不懂l，r，p有什么不同。正确的写着倒是很自然，但是不明白错的为什么错了。。。

//思路： 单点更新，区间查询。
//线段树每个节点记录该区间从左向右最后一个相同的位置，若左边第一个就不相同，则记为-1
//这道题目的难点就是在于区间的合并的时候
#include <iostream>
#include <cstring>
#include <cstdio>
#define lson l,m, rt << 1
#define rson m+ 1, r, rt << 1 | 1
#define maxn 1000100
#define havem int m = (l + r ) >> 1
using namespace std;
int d[maxn << 2],Max;
char s1[maxn],s2[maxn];
void push_up(int rt , int m){
if(d[rt << 1] == m){
d[rt] = max(d[rt << 1] , d[rt << 1 | 1]);
}else
d[rt] = d[rt << 1];
}
void build(int l , int r, int rt){
if(l == r){
if(s1[l] == s2[r]) d[rt] = l;
else d[rt] = -1;    return ;
}
int m = (l + r) >> 1;
build(lson); build(rson);

push_up(rt,m);
}
//单点更新
void update(int id, int p, char cc, int l , int r, int rt){
if(l == r){
//        cout << p << l << r << endl;
if(id == 1) s1[p] = cc;//修改，这里如果是l也错了。难道说递归到底层l，r，p还不一样？好像是一样的啊。。。无语了。。。。
else s2[p] = cc;
if(s1[l] == s2[r]) d[rt] = l;
//这里一定要是l和r，如果是p，就wa了。why？
else d[rt] = -1;    return ;
//永远不要忘记return
}
havem;
if(p <= m) update(id,p,cc,lson);
else update(id,p,cc,rson);
push_up(rt,m);
}
//区间询问
int query(int L, int R, int l , int r, int rt){
if(L == l && R == r) return d[rt];
int m = (l + r) >> 1 , ret1= -1, ret2 = -1;
if(R <= m) return query(L,R,lson);
if(L > m) return query(L,R,rson);

//执行到这里就是要查询的区间[L,R]覆盖[l,r]的中点了，这个时候需要注意了。
ret1 = query(L,m,lson);

if(ret1 == m) {
ret2 = query(m + 1, R, rson);
return max(ret1,ret2);
}
else return ret1;
}
int main(){
int ca,Q,id,p,Max,cas = 1;
char cc[4];
scanf("%d",&ca);
while(ca --){
scanf("%s%s",s1,s2);
scanf("%d",&Q);
int len1 =strlen(s1), len2 = strlen(s2);
Max = min(len1, len2) - 1;

build(0,Max,1);
printf("Case %d:\n",cas++);
while(Q--){
scanf("%d",&id);
if(id == 1){
scanf("%d %d %s",&id,&p,cc);
update(id,p,cc[0],0,Max,1);
}
else{
scanf("%d",&p);
int ans = query(p,Max,0,Max,1);//ans返回的是位置
if(ans == -1) printf("0\n");
else printf("%d\n",ans - p + 1);
}
}
}
return 0;
}