HDU-----(4858)项目管理(模拟)

项目管理

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

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

 

Author

CLJ

 

Source

BestCoder Round #1

 

代码：

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<vector>
using namespace std;
const int maxn=100005;
vector<vector<int> >arr;
//vector<int>arr[maxn];
int ans[maxn];
int n,m,i;
int main(){
  int cas,a,b,q,cmd,u,v;
  scanf("%d",&cas);
  while(cas--){
    scanf("%d%d",&n,&m);
    arr.clear();
    arr.resize(n+1);
    memset(ans,0,sizeof(int)*(n+1));
    for(i=0;i<m;i++){
     scanf("%d%d",&a,&b);
     arr[b].push_back(a);
     arr[a].push_back(b);
    }
    scanf("%d",&q);
   while(q--){
        scanf("%d",&cmd);
    if(cmd==1){
       scanf("%d",&u);
       printf("%d
",ans[u]);
    }
    else{
      scanf("%d%d",&u,&v);
      int len=arr[u].size();
      for(i=0;i<len;i++)
        ans[arr[u][i]]+=v;
    }
   }
   //for(i=1;i<=n;i++)arr[i].clear();
  }
  return 0;
}

优化之后：

//#pragma comment(linker,"STACK:1024000000,1024000000");
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<vector>
using namespace std;
const int maxn=100005;
vector<int>arr[maxn];
int ans[maxn];
int n,m,i;
int main(){
  int cas,a,b,q,cmd,u,v;
  scanf("%d",&cas);
  while(cas--){
    scanf("%d%d",&n,&m);
    memset(ans,0,sizeof(int)*(n+1));
    for(i=0;i<m;i++){
     scanf("%d%d",&a,&b);
     arr[b].push_back(a);
     arr[a].push_back(b);
    }
    scanf("%d",&q);
   while(q--){
        scanf("%d",&cmd);
    if(cmd==1){
       scanf("%d",&u);
       printf("%d
",ans[u]);
    }
    else{
      scanf("%d%d",&u,&v);
      int len=arr[u].size();
      for(i=0;i<len;i++)
        ans[arr[u][i]]+=v;
    }
   }
   //for(i=1;i<=n;i++)arr[i].clear();
  }
  return 0;
}