C++ 가변 배열(Variable Sized Arrays)

개요

C++ Standard Template Library의 vector 사용법을 학습합니다.

문제 출처 : Variable Sized Arrays | HackerRank

문제 요약

주어진 입력을 바탕으로 가변 길이의 배열들을 만들고, Query에 따라 각 배열의 원소를 화면에 출력하라.

예시 입력

2 2                // Array 2개, Query 2건
3 1 5 4            // 0번째 Array는 원소가 3개, 각 원소는 1,5,4
5 1 2 8 9 3        // 1번째 Array는 원소가 5개, 각 원소는 1,2,8,9,3
0 1                // 첫번째 Query 0번째 Array의 1번째 Index의 원소를 출력하라
1 3                // 두번째 Query 1번째 Array의 3번째 Index의 원소를 출력하라

예시 출력

5
9

풀이

#include  <cmath>
#include  <cstdio>
#include  <vector>
#include  <iostream>
#include  <algorithm>

using  namespace std;

int main() {

    int n, q;
    cin >> n >> q;
    vector<vector<int>> A(n);   // n 개의 Array

    for (int i=0;i<n;i++) {
        int k;
        cin >> k;
        A[i] = vector<int>(k);  // K 개의 원소를 가진 Array
        for (int j=0;j<k;j++) {
            int e;
            cin >> e;
            A[i][j] = e;
        }        
    }

    for (int i=0;i<q;i++) {
        int row, col;
        cin >> row >> col;
        cout << A[row][col] << endl;
    }
    return  0;

}