hdu 4467 莫队算法

k=根号（入度出度之和）

与该点相连的点大于等于k为重点，否则为轻点

对轻点进行暴力维护，重点就将他周围的权值用一个二维数组sum[i][j]维护，i表示这个重点它周围的j这个颜色的权值和 

#include <iostream>
#include <algorithm>
#include <vector>
#include <cmath>
#include <stdio.h>
#include <cstring>
using namespace std;
#define N 2300010
#define ll long long
const int maxn=100005;
struct node
{
    int u,v;
    ll w;
    bool operator<(const node &b)const{
        if(u==b.u) return v<b.v;
        return u<b.u;
    }
}e[maxn];
int now,x,y,a1,b,q,n,m,a[maxn],k,du[maxn],sp[maxn],head[maxn<<2][2];
ll ans[3],sum[maxn][2];
struct edge
{
    int to,next;
    ll val;
}g[maxn<<2];
void add(int t,int x,int y,ll v){
	g[++now].to=y,g[now].val=v,g[now].next=head[x][t],head[x][t]=now;
}
int main(){
    int ca=1;
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%lld",&e[i].u,&e[i].v,&e[i].w);
            if(e[i].u>e[i].v) swap(e[i].u,e[i].v);
        }
        sort(e,e+m);
        int cnt=0;
        for(int i=0,j;i<m;i=j)
        {
            for(j=i+1;j<m&&e[i].u==e[j].u&&e[i].v==e[j].v;j++)
                e[i].w+=e[j].w;
            e[cnt++]=e[i];
        }
        k=(int)sqrt(cnt<<1);
        memset(du,0,sizeof(du));
        memset(head,0,sizeof(head));
        memset(ans,0,sizeof(ans));
        memset(sum,0,sizeof(sum));
        now=0;
        for(int i=0;i<cnt;i++) du[e[i].u]++,du[e[i].v]++;
        for(int i=1;i<=n;i++) sp[i]=(du[i]>=k);//判断是否为重点
        for(int i=0;i<cnt;i++)
        {
            x=e[i].u,y=e[i].v;
            ll w=e[i].w;
            if(sp[x]) add(1,y,x,w);
            else add(0,x,y,w);
            if(sp[y]) add(1,x,y,w);
            else add(0,y,x,w);//重点被访问，轻点要访问
        }
        for(int i=0;i<cnt;i++)
        {
            x=e[i].u,y=e[i].v;
            ll w=e[i].w;
            if(sp[x]) sum[x][a[y]]+=w;
            if(sp[y]) sum[y][a[x]]+=w;
            ans[a[x]+a[y]]+=w;
        }
        printf("Case %d:\n",ca++);
        char s[20];
        scanf("%d",&q);
        while(q--)
        {
            scanf("%s",s);
            if(s[0]=='A')
            {
                scanf("%d%d",&a1,&b);
                printf("%lld\n",ans[a1+b]);
            }
            else
            {
                scanf("%d",&x);
                a[x]^=1;
                if(sp[x])
                {
                    for(int i=0;i<=1;i++)
                    {
                        ans[(a[x]^1)+i]-=sum[x][i];
                        ans[a[x]+i]+=sum[x][i];
                    }
                }
                else
                {
                    for(int i=head[x][0];i;i=g[i].next)
                    {
                        ans[(a[x]^1)+a[g[i].to]]-=g[i].val;
                        ans[a[x]+a[g[i].to]]+=g[i].val;
                    }
                }
                for(int i=head[x][1];i;i=g[i].next)
                {
                    sum[g[i].to][a[x]^1]-=g[i].val;
                    sum[g[i].to][a[x]]+=g[i].val;
                }
            }
        }
    }
    return 0;
}