首页 > 代码库 > HDU BestCoder Round #1 1002 项目管理

HDU BestCoder Round #1 1002 项目管理

项目管理


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


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

水题,直接模拟一遍过程就行了。

#include <stdio.h>
#include <vector>
#include <string.h>
#define maxn 100002
using namespace std;

vector<int> vec[maxn];
vector<int>::iterator it;
int val[maxn];

int getVal(int k)
{
    int sum = 0;
    for(it = vec[k].begin(); it != vec[k].end(); ++it)
        sum += val[*it];
    return sum;
}

int main()
{
    int t, n, m, i, a, b, q, com;
    scanf("%d", &t);
    while(t--){
        scanf("%d%d", &n, &m);
        memset(vec, 0, sizeof(vec));
        memset(val, 0, sizeof(val));
        for(i = 1; i <= m; ++i){
            scanf("%d%d", &a, &b);
            vec[a].push_back(b);
            vec[b].push_back(a);
        }
        scanf("%d", &q);
        while(q--){
            scanf("%d%d", &com, &a);
            if(com == 1) printf("%d\n", getVal(a));
            else{
                scanf("%d", &b);
                val[a] += b;
            }
        }
    }
    return 0;
}