spoj375 Query on a tree

贴出代码,其中注释部分大多为自己打的时候的错误,以后回顾时稍微注意一下。

wrong! ——错误

forget!——忽略的

add!——其实这里面的阿得得是毫无意义的修改...(之前不知道错了哪里,就一顿乱改...ovo)


【bling!】

最终错因。

有一个地方把mid+1写成了mid+2........(尴尬

#include
#include
#include
#include
#include

using namespace std;

#define maxn (100000+10)
#define INF 0x7f7f7f7f

struct node{
	int to, next;
}e[maxn*3];
int head[maxn], cnt;

void addedge(int u, int v){
	e[cnt].to = v;
	e[cnt].next = head[u];
	head[u] = cnt++;
}
/*-----------------cut tree----------------------------*/
int dep[maxn], son[maxn], fa[maxn], top[maxn], siz[maxn], line[maxn], sum;//
void dfs1(int u, int pa, int depth){
	dep[u] = depth; fa[u] = pa; son[u] = 0; siz[u] = 1;//
	
	for(int i = head[u]; i != -1; i = e[i].next){
	    int v = e[i].to;
	    if(v == pa) continue;
	    dfs1(v, u, depth+1);
	    siz[u] += siz[v];
	    if(siz[v] > siz[son[u]]) son[u] = v;
	}
}
void dfs2(int u, int pa){
	line[u] = ++sum; top[u] = pa;
	if(son[u] != 0) dfs2(son[u], top[u]);
	for(int i = head[u]; i != -1; i = e[i].next)
	    if(e[i].to != son[u] && e[i].to != fa[u]) 
			dfs2(e[i].to, e[i].to);
}

/*---------------------seg tree------------------------*/
struct nodde{
    int l, r, val, mx;
/*add--*/int mid(){return (l + r) >> 1;}
}tree[maxn*2];

void build(int pos, int l, int r){
    tree[pos].l = l; tree[pos].r = r; tree[pos].mx = tree[pos].val = -INF;

/*forget!*/ if(l == r) return;  
 
//    int mid = tree[pos].l+((tree[pos].r-tree[pos].l) >> 1);
 	int mid = tree[pos].mid();   
/*wrong!    build(pos*2, tree[pos].l, mid);
    build(pos*2+1, mid+1, tree[pos].r);*/

    build(pos*2, l, mid);
    
/*wrong!    build(pos*2+1, mid+2, r)*/    
    
    build(pos*2+1, mid+1, r);
}

void update(int pos, int id, int w){
    if(tree[pos].l == tree[pos].r){
	    tree[pos].mx = tree[pos].val = w;
		return;
	}
//	int mid = tree[pos].l+((tree[pos].r-tree[pos].l) >> 1);
	int mid = tree[pos].mid();
	if(mid >= id) update(pos*2, id, w);
	else update(pos*2+1, id, w);

/*forget!*/tree[pos].mx = max(tree[pos*2].mx, tree[pos*2+1].mx);

}

int query(int pos, int L, int R){
    if(tree[pos].l >= L && tree[pos].r <= R) return tree[pos].mx;
//    int mid = tree[pos].l+((tree[pos].r-tree[pos].l) >> 1);
	int mid = tree[pos].mid();
    int ans = -INF;
    if(mid >= L) ans = max(ans, query(pos*2, L, R));
    if(mid < R) ans = max(ans, query(pos*2+1, L, R));
    
    return ans;
}

int lca(int u, int v){
	int ans = -INF;
	while(top[u] != top[v]){
	    if(dep[top[u]] < dep[top[v]]) swap(u, v);
	    ans = max(ans, query(1, line[top[u]], line[u]));
	    u = fa[top[u]];
	}

/*wrong!	if(dep[u] < dep[v]) swap(u, v);
	ans = max(ans, query(1, line[top[u]]+1, line[u]));*/

	if(dep[u] > dep[v]) swap(u, v);
	if(u != v) ans = max(ans, query(1, line[u]+1, line[v]));
	return ans;
}

int d[maxn][3];
int main(){
/*	freopen("test.in", "r", stdin);
	freopen("test.out","w",stdout);
*/	int _, n;
	scanf("%d", &_);
	
	while(_--){
		cnt = sum = 0;
		memset(head, -1, sizeof(head));
		
		scanf("%d", &n);
		for(int i = 1; i  d[i][1]) swap(d[i][1], d[i][0]);*/
			if(dep[d[i][0]] > dep[d[i][1]]) swap(d[i][0], d[i][1]);		    
		    update(1, /*wrong! d[i][1] */line[d[i][1]], d[i][2]);
		}
		
		char s[10];
		while(1){
			scanf("%s", s);
			if(s[0] == 'D') break;
			if(s[0] == 'Q'){
				int a, b;
				scanf("%d%d", &a, &b);
				printf("%d\n", lca(a, b));
			}
			else{
				int a, b;
				scanf("%d%d", &a, &b);
				update(1, line[d[a][1]], b);
			}
		}
		printf("\n");
	}
	
	return 0;
}



你可能感兴趣的:(日常刷题)