hdu 4858 项目管理（图的分治）

项目管理

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 2193    Accepted Submission(s): 818

Problem Description

我们建造了一个大项目！这个项目有n个节点，用很多边连接起来，并且这个项目是连通的！
两个节点间可能有多条边，不过一条边的两端必然是不同的节点。
每个节点都有一个能量值。

现在我们要编写一个项目管理软件，这个软件呢有两个操作：
1.给某个项目的能量值加上一个特定值。
2.询问跟一个项目相邻的项目的能量值之和。（如果有多条边就算多次，比如a和b有2条边，那么询问a的时候b的权值算2次）。

 

Input

第一行一个整数T(1 <= T <= 3),表示测试数据的个数。
然后对于每个测试数据，第一行有两个整数n(1 <= n <= 100000)和m(1 <= m <= n + 10)，分别表示点数和边数。

然后m行，每行两个数a和b，表示a和b之间有一条边。
然后一个整数Q。

然后Q行，每行第一个数cmd表示操作类型。如果cmd为0，那么接下来两个数u v表示给项目u的能量值加上v(0 <= v <= 100)。
如果cmd为1，那么接下来一个数u表示询问u相邻的项目的能量值之和。

所有点从1到n标号。

 

Output

对每个询问，输出一行表示答案。

 

Sample Input

  
  

   
   1
3 2
1 2
1 3
6
0 1 15
0 3 4
1 1
1 3
0 2 33
1 2
  
  

 

Sample Output

  
  

   
   4
15
15
  
  

思路：直接链式前向星暴力会炸，但是用自带的邻接表却不会炸。。真是玄学。

标准解法是图的分治，本来要把所有边都存下来的，现在先去重，再存在终点度数比起点大的边，这样复杂度就降低很多了。 700ms跑过

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 100050
struct Edge
{
    int v,next,num,last;
}edge[N*4];
struct E
{
    int u,v,num;
}e[N*3];
int cnt,head[N];
int p[N],sum[N],dug[N];
bool cmp(E a,E b)
{
    if(a.u==b.u) return a.v<b.v;
    return a.u<b.u;
}
void addedge(int u,int v,int num)
{
    edge[cnt].v=v;
    edge[cnt].next=head[u];
    edge[cnt].num=num;
    head[u]=cnt++;
}
int main()
{
    int T,n,m,t;
    int u,v,Q,cnt1;
    scanf("%d",&T);
    while(T--)
    {
        cnt=0;
        memset(head,-1,sizeof(head));
        memset(sum,0,sizeof(sum));
        memset(dug,0,sizeof(dug));
        memset(p,0,sizeof(p));
        scanf("%d %d",&n,&m);
        for(int i=1;i<=m;i++)
        {
            scanf("%d %d",&u,&v);
            if(u>v) swap(u,v);
            e[i].u=u,e[i].v=v,e[i].num=1;
        }
        sort(e+1,e+m+1,cmp);
        int tot=1;
        for(int i=2;i<=m;i++)
        {
            if(e[i].u==e[i-1].u&&e[i].v==e[i-1].v) e[tot].num++;
            else e[++tot]=e[i];
        }

        for(int i=1;i<=tot;i++)
            dug[e[i].u]++,dug[e[i].v]++;
        for(int i=1;i<=tot;i++)
        {
            if(dug[e[i].v]<dug[e[i].u]) addedge(e[i].v,e[i].u,e[i].num);
            else addedge(e[i].u,e[i].v,e[i].num);
        }
        scanf("%d",&Q);
        while(Q--)
        {
            scanf("%d",&t);
            if(t==0)
            {
                scanf("%d %d",&u,&v);
                p[u]+=v;
                for(int i=head[u];i!=-1;i=edge[i].next)
                {
                    int w=edge[i].v;
                    sum[w]+=edge[i].num*v;
                }
            }
            else
            {
                scanf("%d",&u);
                int ans=sum[u];
                for(int i=head[u];i!=-1;i=edge[i].next)
                {
                    v=edge[i].v;
                    if(edge[i].last!=p[v])
                    {
                        ans+=edge[i].num*(p[v]-edge[i].last);
                        edge[i].last=p[v];
                    }
                }
                sum[u]=ans;
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}