HDU - 4858:项目管理(vector)

题目链接
我们建造了一个大项目!这个项目有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
使用暴力解法,但是无法定义一个100000大小的二维数组,所以想到使用vector容器,来代替二维数组
先来看一下vector的使用和下标的访问

#include 
using namespace std;
//定义一个相当于一维数组的容器
vector <int>b;
//定义一个相当于二维数组的容器
vector <int>a[10];
int main()
{
    //这种存储方式相当于一位数组
    for(int i = 0;i < 10;i++)
        b.push_back(i);
    //这种存储方式相当于二维数组
    for(int i = 0;i < 10;i++)
        a[i].push_back(i);
    for(int i = 0;i < 10;i++)
        printf("%d ",b[i]);
    printf("\n");
    for(int i = 0;i < 10;i++)
        printf("%d ",a[i][0]);
    return 0;
}

下面是本题代码

#include 
#include 
#include 
#include 
#include 
#define N 100100

using namespace std;

vector <int>g[N];

int c[N];

int main()
{
    int T,a,b,n,m,q,op,i;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d %d",&n,&m);
        for(i = 0;i <= n;i++)           //清空容器
            g[i].clear();
        while(m--)
        {
            scanf("%d %d",&a,&b);
            g[b].push_back(a);          //输入数据,类似于一个二维数组
            g[a].push_back(b);
        }
        scanf("%d",&q);
        memset(c, 0, sizeof(c));
        while(q--)
        {
            scanf("%d",&op);
            if(op == 0)
            {
                scanf("%d %d",&a,&b);
                c[a] += b;
            }
            else
            {
                int s = 0;
                scanf("%d",&a);
                for(i = 0;i < g[a].size();i++)           
                {
                    s += c[g[a][i]];		//访问方式也与二维数组相似
                }
                printf("%d\n",s);
            }
        }
    }
    return 0;
}

你可能感兴趣的:(STL)