백준 10431번 줄세우기 C++

출처 : https://www.acmicpc.net/problem/10431

 

입력받은 순서대로 줄을 세우는데, 작은 수가 앞에 오도록 줄을 세워야 한다. 이때, 이동이 얼마나 일어나는지를 구하는 문제이다.

 

1. 자신보다 큰 수를 찾는다.

2. 없으면 push_back을 한다.

3. 있으면 중간에 끼워넣고 그 뒤에 있는 수만큼 sum을 증가시킨다.

 

find_if와 람다를 사용해서 문제를 풀어보았다.

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>

using namespace std;
using ll = long long;

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	int p;
	cin >> p;
	while (p--) {
		int t;
		cin >> t;
		int sum = 0;
		vector<int> v;
		for (int i = 0; i < 20; i++) {
			int num;
			cin >> num;
			auto it = find_if(v.begin(), v.end(), [=](int val) {return num < val; });
			if (it == v.end()) v.push_back(num);
			else {
				sum += v.end() - it;
				v.insert(it, num);
			}
		}
		cout << t << " " << sum << "\n";
	}
};