CF176E Archaeology（set用法提示）

题目大意：

给一棵树，每次激活或熄灭一个点，每次问这些点都联通起来所需的最小总边权

分析：

若根据dfs序给所有点排序，为$v1,v2,v3....vk$，那么答案就是$(dis(v1,v2)+dis(v2,v3)+...+dis(vk-1,vk)+dis(vk,v1))/2$

只需要动态的维护这个序列，每次拿出前后两个点后用lca修改答案即可

这道题主要是想学习一下set在这方面的使用

毕竟现在开O2的题比比皆是，要是能少写个平衡树岂不美哉（我也不会写233

我们就默认使用的是c++11的标准吧

---

如何为set重载运算符

首先你要有一个结构体，并在里面重载一个bool类型的（）函数，先这样（这里以将数字按d数组的大小排序为例）：

struct cmp{bool operator () (const int &a,const int &b){return d[a]<d[b];}};

接下来这样声明set：

set<int,cmp>S;

如何查找一个元素的前驱后继等

我们这里声明迭代器时使用c++11特有的auto用法，看起来方便不少（set<int>::iterator)

这里以查找x的前驱后继为例（循环式的，即若x为最后一个数，后继就是第一个数，反过来同理

auto it=S.lower_bound(x),a=it,b=it;
int l=(it==S.begin()?*--S.end():*--a);
int r=(it==--S.end()?*S.begin():*++b);

注意这里end()函数返回的是个超尾，所以要--

这里我们看到迭代器的移动用加减即可，取值时用*即可

插入删除

insert和erase，千万别记错了

S.insert(x);
S.erase(x);

---

接下来放上这道题的代码

 #include<bits/stdc++.h>
 #define N 100005
 #define ll long long
 using namespace std;
 int n,q;
 int h[N],to[*N],nxt[*N],w[*N],etop;
 void add(int a,int b,int c){to[++etop]=b,nxt[etop]=h[a],w[etop]=c,h[a]=etop;}
 int fa[N][],d[N],tot,dep[N];
 ll len[N][];
 void dfs(int u){
     d[u]=++tot;
     for(int i=;i<=;i++){
         fa[u][i]=fa[fa[u][i-]][i-];
         len[u][i]=len[u][i-]+len[fa[u][i-]][i-];
         if(fa[u][i]==)break;
     }
     for(int k=h[u],v=to[k];k;k=nxt[k],v=to[k])
     if(v!=fa[u][]){
         fa[v][]=u;len[v][]=w[k];
         dep[v]=dep[u]+;
         dfs(v);
     }
 }
 ll LCA(int x,int y){
     ll ans=;
     if(dep[x]<dep[y])swap(x,y);
     for(int i=;i>=;i--)
     if(fa[x][i]&&dep[fa[x][i]]>=dep[y]){
         ans+=len[x][i];
         x=fa[x][i];
     }
     if(x==y)return ans;
     for(int i=;i>=;i--)
     if(fa[x][i]!=fa[y][i]){
         ans+=len[x][i];
         ans+=len[y][i];
         x=fa[x][i];
         y=fa[y][i];
     }
     ans+=len[x][];
     ans+=len[y][];
     return ans;
 }
 struct cmp{bool operator () (const int &a,const int &b){return d[a]<d[b];}};
 set<int,cmp>S;
 ll ans;
 void ins(int x){
     S.insert(x);
     auto it=S.lower_bound(x),a=it,b=it;
     int l=(it==S.begin()?*--S.end():*--a);
     int r=(it==--S.end()?*S.begin():*++b);
     ans-=LCA(l,r);
     ans+=LCA(l,x);
     ans+=LCA(x,r);
 }
 void del(int x){
     auto it=S.lower_bound(x),a=it,b=it;
     int l=(it==S.begin()?*--S.end():*--a);
     int r=(it==--S.end()?*S.begin():*++b);
     ans+=LCA(l,r);
     ans-=LCA(l,x);
     ans-=LCA(x,r);
     S.erase(x);
 }
 char o[];
 int main(){
     scanf("%d",&n);
     for(int i=,a,b,c;i<n;i++){
         scanf("%d%d%d",&a,&b,&c);
         add(a,b,c),add(b,a,c);
     }
     dfs();
     scanf("%d",&q);
     while(q--){
         scanf("%s",o);
         if(o[]=='?')cout<<ans/<<endl;
         else{
             int x;scanf("%d",&x);
             if(o[]=='+')ins(x);
             else del(x);
         }
     }
     return ;
 }

---

2019.6.18 update

今天在写一个扫描线题时使用上面的方法重置set的比较符出现了问题，使用lower_bound会在大数据下WA掉，但upper_bound则没问题

如果使用常规的重载运算符的话则两种方式都没问题……（蒙圈ing

这里给个当时的两种比较函数吧

这是出了问题的：

struct cmp{
    bool operator () (const hu &A,const hu &B){
        double as=A.y+sqrt(A.r*A.r-(A.x-X)*(A.x-X))*(1.0*A.u);
        double bs=B.y+sqrt(B.r*B.r-(B.x-X)*(B.x-X))*(1.0*B.u);
        if(as+eps>bs&&bs+eps>as)return A.u<B.u;
        else return as<bs;
    }
};

这个是改成重载运算符的：

bool operator < (const hu &A,const hu &B){
    double as=A.y+sqrt(A.r*A.r-(A.x-X)*(A.x-X))*(1.0*A.u);
    double bs=B.y+sqrt(B.r*B.r-(B.x-X)*(B.x-X))*(1.0*B.u);
    if(as+eps>bs&&bs+eps>as)return A.u<B.u;
    else return as<bs;
}

如果哪位大神路过希望能指点一下555（；´д｀）ゞ

还有NOI没有C++11

所以跟我一起拼一遍

iterator

再来3遍

iterator

iterator

iterator

ojbk!