萌新10分代码求调,感谢各位大佬帮助
查看原帖
萌新10分代码求调,感谢各位大佬帮助
1344956
BestCNSoda楼主2024/11/26 22:00

测了多组大数据,在几千或几万行处突然输出-1,不知道哪里出错

#include<bits/stdc++.h>
#define lx x<<1
#define rx x<<1|1
#define int long long
using namespace std;
int tree[200001],minn[200001];
int n,q,cnt,dfn;
int siz[100001],id[100001],deep[100001],head[100001],top[100001],fa[100001],son[100001];
int w[100001],w1[100001];
struct Edge {int to,nex;}edge[100001];
void pushup(int x)
{
    tree[x]=tree[lx]+tree[rx];
    minn[x]=max(minn[lx],minn[rx]);
}
void build(int x,int l,int r)
{
    if(l==r)
    {
        tree[x]=w1[l];
        minn[x]=w1[l];
        return;
    }
    int mid=(l+r)>>1;
    build(lx,l,mid);
    build(rx,mid+1,r);
    pushup(x);
}
void update(int ll,int x,int l,int r,int z)
{
    if(l==r&&l==ll)
    {
           minn[x]=z;
           tree[x]=z;
           return;
    }
    int mid=(l+r)>>1;
    if(ll<=mid) update(ll,lx,l,mid,z);
    if(ll>mid) update(ll,rx,mid+1,r,z);
    pushup(x);
}
int query(int ll,int rr,int x,int l,int r)
{
    if(ll<=l&&r<=rr)
    {
        return tree[x];
    }
    int res=0;
    int mid=(l+r)>>1;
    if(ll<=mid) res+=query(ll,rr,lx,l,mid);
    if(rr>mid) res+=query(ll,rr,rx,mid+1,r);
    return res;
}
int querymin(int ll,int rr,int x,int l,int r)
{
    if(ll<=l&&r<=rr)
    {
        return minn[x];
    }
    int small=-1;
    int mid=(l+r)>>1;
    if(ll<=mid) small=max(querymin(ll,rr,lx,l,mid),small);
    if(rr>mid) small=max(querymin(ll,rr,rx,mid+1,r),small);
    return small;
}
void addedge(int u,int v)
{
    edge[++cnt].to=v;
    edge[cnt].nex=head[u];
    head[u]=cnt;
}
void dfs(int x,int fath)
{
    siz[x]=1;
    deep[x]=deep[fath]+1;
    fa[x]=fath;
    for(int i=head[x];i;i=edge[i].nex)
    {
        int v=edge[i].to;
        if(v==fath) continue;
        dfs(v,x);
        siz[x]+=siz[v];
        if(!son[x]||siz[son[x]]<siz[v]) son[x]=v;
    }
}
void dfs1(int x,int topx)
{
      id[x]=++dfn;
      top[x]=topx;
      w1[dfn]=w[x];
      if(!son[x]) return;
      dfs1(son[x],topx);
      for(int i=head[x];i;i=edge[i].nex)
      {
          int v=edge[i].to;
          if(v!=fa[x]&&v!=son[x])
          {
              dfs1(v,v);
          }
      }
}
int query_range(int x,int y)
{
    int res=0;
    while(top[x]!=top[y])
    {
        if(deep[top[x]]<deep[top[y]]) swap(x,y);
        res+=query(id[top[x]],id[x],1,1,n);
        x=fa[top[x]];
    }
    if(deep[x]>deep[y]) swap(x,y);
    res+=query(id[x],id[y],1,1,n);
    return res;
}
int querym(int x,int y)
{
    int res=-1;
    while(top[x]!=top[y])
    {
        if(deep[top[x]]<deep[top[y]]) swap(x,y);
        res=max(querymin(id[top[x]],id[x],1,1,n),res);
        x=fa[top[x]];
    }
    if(deep[x]>deep[y]) swap(x,y);
    res=max(querymin(id[x],id[y],1,1,n),res);
    return res;
}
signed main()
{
    //freopen("count2.in","r",stdin);
    //freopen("count.out","w",stdout);
    cin>>n;
    for(int i=1;i<n;i++)
    {
        int u,v;
        scanf("%lld%lld",&u,&v);
        addedge(u,v);
        addedge(v,u);
    }
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%lld",&x);
        w[i]=x;
    }
    dfs(1,1);
    dfs1(1,1);
    build(1,1,n);
    cin>>q;
    while(q--)
    {
        string s;
        cin>>s;
        int x,y;
        scanf("%lld%lld",&x,&y);
        //cout<<x<<' '<<y<<endl;
        if(s=="QMAX")
        {
            printf("%lld\n",querym(x,y));
            //cout<<querymax(x,y)<<endl;
        }
        if(s=="QSUM")
        {
            printf("%lld\n",query_range(x,y));
        }
        if(s=="CHANGE")
        {
            update(id[x],1,1,n,y);
        }
    }
    return 0;
}

2024/11/26 22:00
加载中...