HDU 4858 项目管理

传送门：http://acm.hdu.edu.cn/showproblem.php?pid=4858

题意：

　　给一张图（稀疏图）且图上可能有重边，每个点都有一个value且初始都为0，有两个操作1、使某个点的value加上一个值；2、求一个点所连的所有边的另一个端点的value和，如果有重边则计算多次。

题解：

　　通过分块的思想，把图上的点按连边的多少(是否小于sqrt(m))分成重点和轻点。将所有轻点完全建边，重点只建与重点之间的边。每次更新点的value时更新所有连接的边的sum因为每个点连的边不超过sqrt(m)所以更新的复杂度是sqrt(m)，询问时对于轻点，暴力计算与它连的点的value和，复杂度为sqrt(m)，对于重点只需返回该点的sum即可。

 

#include <bits/stdc++.h>
using namespace std;
const int maxn=100000+200;
vector<int> g[maxn];
struct Edge {
    int u,v;
}e[maxn*2];
int T,Q,n,m,u,unit,v,op,ranks[maxn],sum[maxn],a[maxn];

void update(int x, int d) {
    a[x]+=d;
    for(auto i:g[x]) sum[i]+=d;
}
int query(int x) {
    if(ranks[x]<unit) {
        sum[x]=0;
        for(auto i:g[x]) sum[x]+=a[i];
        return sum[x];
    }
    else return sum[x];
}
int main() {
    scanf("%d",&T);
    while(T--) {
        scanf("%d%d",&n,&m);
        memset(sum,0,sizeof(sum));
        memset(a,0,sizeof(a)); 
        memset(ranks,0,sizeof(ranks));
        for(int i=1;i<=m;++i) {
            scanf("%d%d",&e[i].u,&e[i].v);
            ranks[e[i].u]++;
            ranks[e[i].v]++;
        }
        unit=sqrt(m+0.5);
        for(int i=1;i<=m;++i) {
            u=e[i].u; v=e[i].v;
            if(ranks[u]<unit) g[u].push_back(v);
            else if(ranks[v]>=unit) g[u].push_back(v);
            if(ranks[v]<unit) g[v].push_back(u);
            else if(ranks[u]>=unit) g[v].push_back(u);
        }
        scanf("%d",&Q);
        while(Q--) {
            scanf("%d%d",&op,&u);
            if(op) printf("%d\n",query(u));
            else {
                scanf("%d",&v); update(u,v);
            }
        }
        for(int i=1;i<=n;++i) g[i].clear();
    }
    return 0;
}