ZOJ3742Bellywhite's Algorithm Homework

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3742

　　给定n个点m条边的无向图，支持两个操作：

　　　　C x：将与x相邻的边符号取反。

　　　　Q x：

　　　　　　如果x为'+'，输出正边权和。

　　　　　　如果x为'-'，输出负边权和。

　　　　　　如果x为'A'，输出总边权和。

　　按度数分块，将度数≥√m的点记为重点，否则为轻点。

　　对于重点，我们记录positive_sum和negative_sum，代表相邻的轻-重边正、负边权和。并且记录与之相邻的重点。

　　进行修改操作的时候：

　　　　对于重点的取反，我们打上opp标记，修改时取反并更新一下即可。注意重-重边要单独修改。

　　　　对于轻点的取反，我们暴力修改，注意轻-重边的实际边权是opp*边权即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long int64;
typedef unsigned int uint;
const int maxn=50015,maxm=100015;
int n,m,q;
vector<int> w[maxn];
bool heavy[maxn],opp[maxn];
int64 res_p,res_n,sum_p[maxn],sum_n[maxn];
int tot,deg[maxn],now[maxn],pre[maxm],son[maxm],val[maxm];
void connect(int u,int v,int w){pre[++tot]=now[u];now[u]=tot;son[tot]=v;val[tot]=w;}
void initial(){
    res_p=res_n=0;tot=1;
    memset(opp,0,sizeof(opp));
    memset(now,0,sizeof(now));
    memset(deg,0,sizeof(deg));
    memset(heavy,0,sizeof(heavy));
    memset(sum_p,0,sizeof(sum_p));
    memset(sum_n,0,sizeof(sum_n));
    for (int i=1;i<=n;++i) w[i].clear();
}
void init(int cases){
    initial();
    if (scanf("%d%d%d",&n,&m,&q)==EOF) exit(0);
    if (cases>1) printf("\n");
    for (int x,y,v,i=1;i<=m;++i){
        scanf("%d%d%d",&x,&y,&v);
        connect(x,y,v);connect(y,x,v);
        (v>0?res_p:res_n)+=v;++deg[x];++deg[y];
    } 
    for (int lim=sqrt(m),i=1;i<=n;++i) heavy[i]=(deg[i]>lim);
    for (int i=1;i<=n;++i)
        for (int p=now[i];p;p=pre[p])
            if (heavy[i]&&heavy[son[p]]) w[i].push_back(p);
            else (val[p]>0?sum_p[i]:sum_n[i])+=val[p];
}
void query(){
    char op[5];scanf("%s",op);
    switch (op[0]){
        case 'A':printf("%lld\n",res_p+res_n);break;
        case '-':printf("%lld\n",res_n);break;
        case '+':printf("%lld\n",res_p);break;
    }
}
void opposite(int p){
    (val[p]>0?res_p:res_n)-=val[p];
    val[p]=val[p^1]=-val[p];
    (val[p]>0?res_p:res_n)+=val[p];
}
void modify_heavy(int u){
    opp[u]^=1;
    res_p-=sum_p[u];res_n-=sum_n[u];
    swap(sum_p[u],sum_n[u]);sum_p[u]=-sum_p[u];sum_n[u]=-sum_n[u];
    res_p+=sum_p[u];res_n+=sum_n[u];
    for (uint i=0;i<w[u].size();++i) opposite(w[u][i]);
}
void modify_light(int u){
    for (int p=now[u];p;p=pre[p]){
        if (heavy[son[p]]){
            int op=opp[son[p]]?-1:1;
            if (op*val[p]>0){
                res_p-=op*val[p];res_n+=op*-val[p];
                sum_p[son[p]]-=op*val[p];sum_n[son[p]]+=op*-val[p];
            }
            else{
                res_n-=op*val[p];res_p+=op*-val[p];
                sum_n[son[p]]-=op*val[p];sum_p[son[p]]+=op*-val[p]; 
            }
            val[p]=val[p^1]=-val[p];
        }
        else opposite(p);
    }
}
void modify(){
    int x;scanf("%d",&x);
    if (heavy[x]) modify_heavy(x);
    else modify_light(x);
}
void work(){
    for (int i=1;i<=q;++i){
        char op[5];scanf("%s",op);
        switch (op[0]){
            case 'Q':query();break;
            case 'C':modify();break;
        }
    }
}
int cases;
int main(){
    while (1){init(++cases);work();}
    return 0;
}