HDU 4467 分块

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4467

题意：给定n个点m条边的无向图，点被染色(黑0/白1)，边带边权。然后q个询问。询问分为两种： Change u:把点u的颜色反转(黑变白，白变黑)，Asksum a b(a,b的值为0/1):统计所以边的权值和,边的两个点满足一个点的颜色为a,一个点的颜色为b。

思路：考虑暴力的做法。修改可以做法O(1),但是查询就得O(m).所以总复杂度为O(m*q)会TLE。然后考虑图分块。参考HDU 4858的做法，将点分为重点和轻点。重点(度数>=sqrt(m))，轻点(度数sqrt(m))。 由于此题查询比较大，所以需要预处理一下优化之后的运算。我们定义每个顶点维护3个属性:顶点的颜色color,与该点相连的边并且另外一个点是白点的边权和W, 同理B是黑色点的边权和。 然后维护3个变量ansWW:边上两个点都是白色的边权和。同理ansWB,ansBB。

修改：

轻点更新自己的color，W,B。同时更新所有的邻点的W,B值

重点更新自己的color，W,B。同时只更新相邻重点的W,B值（所以重点不需要连边到轻点）

对于更新。若未更新时该点的颜色为白色，那么更新后该点的颜色为黑色。那么对于相邻的点的W就要减去对应边的权值，相邻点的B就要加上对应边的权值。

对于更新。若未更新时该点的颜色为白色，那么更新后该点的颜色为黑色。那么边为WW的答案就要减去与该点相连的W的总和，边WB的答案就要减去与该点相邻的B的总和。  并且边为BB的答案就要加上与该点相邻的B的总和，边为WB的答案就要加上与该点相邻的W的总和。

查询：

直接输出对应的值

性质：

与重点相邻的重点不超过sqrt(m)个。

与轻点相邻的所有点不超过sqrt(m)个。

注意：

该题会有重边，如果不合并重边的话会TLE。所以用个map来合并下重边的值即可。

#define _CRT_SECURE_NO_DEPRECATE
#include<stdio.h>  
#include<string.h>  
#include<cstring>
#include<algorithm>  
#include<queue>  
#include<math.h>  
#include<time.h>
#include<vector>
#include<iostream>
#include<map>
using namespace std;
typedef long long int LL;
const int MAXN = 100000 + 10;
struct Edges{
    int u, v; LL w;
    Edges(int u = 0, int v = 0,LL w=0) :u(u), v(v),w(w){};
};vector<Edges>G[MAXN];
map<pair<int, int>, LL>edge;
struct Node{
    int val;
    LL W, B;
    Node(int val = 0, LL W = 0, LL B = 0) :val(val), W(W), B(B){};
}node[MAXN];
LL ansWB, ansWW, ansBB;
int du[MAXN], block;
void init(int n,int m){
    edge.clear(); ansWB = ansWW = ansBB = 0;
    block = (int)sqrt(m + 0.5);
    for (int i = 1; i <= n; i++){
        G[i].clear(); du[i] = 0; node[i].W = node[i].B = 0;
    }
}
void makeGraph(int n, int m){
    for (map<pair<int,int>,LL>::iterator it=edge.begin(); it!=edge.end(); it++){
        int u = it->first.first, v = it->first.second; LL w = it->second;
        if (du[u] >= block&&du[v] >= block){
            G[u].push_back(Edges(u, v, w)); G[v].push_back(Edges(v, u, w));
        }
        if (du[u] < block){
            G[u].push_back(Edges(u, v, w));
        }
        if (du[v] < block){
            G[v].push_back(Edges(v, u, w));
        }
        if (node[u].val&&node[v].val){
            ansWW += w; node[u].W += w; node[v].W += w;
        }
        else if (!node[u].val&&!node[v].val){
            ansBB += w; node[u].B += w; node[v].B += w;
        }
        else{
            ansWB += w; 
            if (node[u].val){
                node[u].B += w; node[v].W += w;
            }
            else{
                node[u].W += w; node[v].B += w;
            }
        }
    }
}
void modify(int pos){
    if (node[pos].val){
        ansWW -= node[pos].W; ansWB -= node[pos].B;
        ansBB += node[pos].B; ansWB += node[pos].W;
        for (int i = 0; i < G[pos].size(); i++){
            node[G[pos][i].v].W -= G[pos][i].w;
            node[G[pos][i].v].B += G[pos][i].w;
        }
    }
    else{
        ansBB -= node[pos].B; ansWB -= node[pos].W;
        ansWW += node[pos].W; ansWB += node[pos].B;
        for (int i = 0; i < G[pos].size(); i++){
            node[G[pos][i].v].W += G[pos][i].w;
            node[G[pos][i].v].B -= G[pos][i].w;
        }
    }
    node[pos].val = !node[pos].val;
}
LL query(int u, int v){
    if (u&&v){ return ansWW; }
    if (!u&&!v){ return ansBB; }
    return ansWB;
}
int main(){
    //#ifdef kirito
    //    freopen("in.txt", "r", stdin);
    //    freopen("out.txt", "w", stdout);
    //#endif
    int n, m, q,Case=1;
    while (~scanf("%d%d", &n, &m)){
        init(n, m);
        for (int i = 1; i <= n; i++)
            scanf("%d", &node[i].val);
        for (int i = 1; i <= m; i++){
            int u, v; LL w; scanf("%d%d%I64d", &u, &v, &w);
            if (u > v){ swap(u, v); }
            if (edge.find(make_pair(u, v)) == edge.end()){
                du[u]++; du[v]++;
                edge.insert(make_pair(make_pair(u, v), w));
            }
            else{
                edge[make_pair(u, v)] += w;
            }
        }
        makeGraph(n, m);
        printf("Case %d:
", Case++);
        scanf("%d", &q);
        while (q--){
            int u,v;  char type[20];
            scanf("%s", type);
            if (type[0]=='A'){
                scanf("%d%d", &u, &v);
                printf("%I64d
", query(u, v));
            }
            else{
                scanf("%d", &u); modify(u);
            }
            //Debug:printf("BB=%I64d WB=%I64d WW=%I64d
", ansBB, ansWB, ansWW);
        }
    }
    return 0;
}