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
Solution
分桶法,把度数不小于sqrt(m)的点称为重点,其他点称为轻点,那么重点总数不超过sqrt(m)个,首先去掉所有重点到轻点的边,每次对于轻点的查询直接暴力,复杂度O(sqrt(m));对于点的更新直接将其价值的增长值加到所有其连接的点上去,那么重点的价值每次查询可以O(1)的给出,由于已经去掉了重点到轻点的边,故每次更新的复杂度也至多是O(sqrt(m))的,总时间复杂度O(q*sqrt(m))
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 111111
int T,n,m,q,e[maxn][2],v[maxn],du[maxn];
ll ans[maxn];
vector<int>g[maxn];
int main()
{
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&m);
memset(v,0,sizeof(v));
memset(du,0,sizeof(du));
memset(ans,0,sizeof(ans));
for(int i=1;i<=n;i++)g[i].clear();
int t=sqrt(m+0.5);
for(int i=1;i<=m;i++)
{
scanf("%d%d",&e[i][0],&e[i][1]);
du[e[i][0]]++,du[e[i][1]]++;
}
for(int i=1;i<=m;i++)
{
int u=e[i][0],v=e[i][1];
if(du[u]>du[v])swap(u,v);
if(du[u]>=t)g[u].push_back(v),g[v].push_back(u);
else
{
g[u].push_back(v);
if(du[v]<t)g[v].push_back(u);
}
}
scanf("%d",&q);
while(q--)
{
int op,u,x;
scanf("%d%d",&op,&u);
if(op==0)
{
scanf("%d",&x);
v[u]+=x;
for(int i=0;i<g[u].size();i++)ans[g[u][i]]+=x;
}
else
{
if(du[u]>=t)printf("%lld\n",ans[u]);
else
{
ll temp=0;
for(int i=0;i<g[u].size();i++)temp+=v[g[u][i]];
printf("%lld\n",temp);
}
}
}
}
return 0;
}