증가하는 시퀀스로 값 채우기

 

보통은 이렇게 쓴다.

 

#include <iostream>
#include <vector>

using namespace std;

void print(vector<int> v);

void main()
{
    vector<int> v(5);
    
    for (int i = 0; i < 5; ++i)
        v[i] = i;
        
    print(v);
}

 

출력

0 1 2 3 4

 

더 쉬운 방법이 있다.

 

#include <iostream>
#include <vector>
#include <algorithm>
#include <numeric>

using namespace std;

void print(vector<int> v);

void main()
{
    vector<int> v(5);
    
    int i = 0;
    generate(v.begin(), v.end(), [&i]() { return i++; });
    print(v);

    generate(v.begin(), v.end(), [i]() mutable { return i++; });
    print(v);
    
    iota(v.begin(), v.end(), 10);
    print(v);    
}

 

출력

0 1 2 3 4
5 6 7 8 9
10 11 12 13 14