HDU 4858 项目管理(分桶法)

来源:互联网 发布:wind资讯金融终端 mac 编辑:程序博客网 时间:2024/06/10 11:38

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 111111int 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;}
0 0
原创粉丝点击