项目管理
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 252 Accepted Submission(s): 88
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
思路:
如果直接暴力,原则上是要TLE的,可惜这题数据比较水~
下面说说我的解法
每个点维护两个值,w[i]-该点的值、sum[i]-相邻节点值的和。
怎样在复杂度较小的范围内维护这两个值呢。
如果把与x相连的点分为两类,一类的度数大于它(该类不超过(2m)^(1/2)),另一类度数小于它。更新和查询时时只操作度数大于它的点,就能降低复杂度了。
一个点更新时度数大于它的相邻点是直接更新了的,查询大点时答案已经统计了,但是查询小点的时候答案这没有统计完,对于小点,这个点又为小点的大点,于是再遍历一下,将未更新的点更新维护sum数组即可。
复杂度就为 q*sqrt(m)了。
还有注意要将重边合并。
代码:
#include <iostream> #include <cstdio> #include <cstring> #include <queue> #include <algorithm> #define maxn 100005 #define MAXN 300005 using namespace std; int n,m,ans,cnt,tot; int head[maxn],dug[maxn]; int sum[maxn],w[maxn]; struct node { int u,v,num; } edge[MAXN]; struct Node { int v,next,num,last; } g[MAXN]; void addedge(int u,int v,int num) { cnt++; g[cnt].v=v; g[cnt].num=num; g[cnt].next=head[u]; head[u]=cnt; } bool cmp(node xx,node yy) { if(xx.u!=yy.u) return xx.u<yy.u ; if(xx.v!=yy.v) return xx.v<yy.v; } int main() { int i,j,t,u,v; scanf("%d",&t); while(t--) { scanf("%d%d",&n,&m); for(i=1; i<=n; i++) dug[i]=head[i]=w[i]=sum[i]=0; for(i=1; i<=m; i++) { scanf("%d%d",&u,&v); if(u>v) swap(u,v); edge[i].u=u; edge[i].v=v; edge[i].num=1; } sort(edge+1,edge+m+1,cmp); tot=1; for(i=2; i<=m; i++) // 合并相同边 { if(edge[i].u==edge[tot].u&&edge[i].v==edge[tot].v) edge[tot].num++; else edge[++tot]=edge[i]; } for(i=1; i<=tot; i++) // 计算度数 { dug[edge[i].u]++; dug[edge[i].v]++; } cnt=0; for(i=1; i<=tot; i++) // 点与相邻节点度数较大的点建图 { u=edge[i].u; v=edge[i].v; if(dug[v]>dug[u]) addedge(u,v,edge[i].num); else addedge(v,u,edge[i].num); } int q,op,val; scanf("%d",&q); while(q--) { scanf("%d",&op); if(op==0) { scanf("%d%d",&u,&val); w[u]+=val; for(i=head[u]; i; i=g[i].next) // 更新大点的sum值 { v=g[i].v; sum[v]+=g[i].num*val; } } else { scanf("%d",&u); ans=sum[u]; for(i=head[u]; i; i=g[i].next) // 计算未更新的大点的值 { v=g[i].v; if(g[i].last!=w[v]) { ans+=g[i].num*(w[v]-g[i].last); g[i].last=w[v]; } } sum[u]=ans; printf("%d\n",ans); } } } return 0; }
hdu 4858 项目管理 (图的分治)