1966 프린터 큐

뭔가 규칙이 있을 거라 생각하고 코드 짰는데 틀렸다ㅋㅋㅋ

 

다시 확인해보니 예외사항이 많아서 그냥 우선순위 큐 쓰기로

 

 

 

 

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <functional>
#include <string>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <cmath>

#define xx first
#define yy second
#define all(x) (x).begin(), (x).end()

using namespace std;
using i64 = long long;
using ii = pair<int, int>;
using ii64 = pair<i64, i64>;

void solve()
{
    int n, m;
    scanf("%d %d", &n, &m);

    vector<int> v(n);
    for (int i = 0; i < n; i++)
        scanf("%d", &v[i]);

    int ans = 0;
    int back = -1;
    int same_back = 0;
    for (int i = 0; i < n; i++)
    {
        
        if (v[i] == v[m] && i < m)
            ans++;
        if (v[i] == v[m] && i > m && back != -1)
            same_back++;
        if (v[i] > v[m])
            ans++;
        if (v[i] > v[m] && i > m && back == -1)
            back = i;
    }

    ans += same_back;

    cout << ans + 1 << endl;
}

int     main()
{
    int t;
    scanf("%d", &t);

    for (int i = 0; i < t; i++)
        solve();
    
    return 0;
}

 

위는 틀린코드

 

 

---

처음에 우선순위 큐 하나만 쓰려고 했는데 이러면 같은 숫자일 때를 제대로 처리할 수 없다.

 

그래서 큐랑 우선순위 큐를 둘 다 두고 우선순위 큐는 정답지 큐는 프린터 동작과정이라 두고 풀었다.

 

프린터랑 큐랑 같으면 pop하고 다르면 다시 뒤로 넣어준다.

 

 

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <functional>
#include <string>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <cmath>

#define xx first
#define yy second
#define all(x) (x).begin(), (x).end()

using namespace std;
using i64 = long long;
using ii = pair<int, int>;
using ii64 = pair<i64, i64>;

void solve()
{
    int n, m;
    scanf("%d %d", &n, &m);

    int search;
    priority_queue<int> pq;
    queue<ii> now;
    for (int i = 0; i < n; i++)
    {
        int in;
        scanf("%d", &in);

        if (i == m)
            search = in;
        pq.push(in);
        now.emplace(in, i);
    }

    int ans = 0;
    while (!pq.empty())
    {
        int top_pq = pq.top();
        ii top_q = now.front();
        now.pop();

        if (top_pq != top_q.first)
        {
            now.push(top_q);
            continue;
        }

        if (search == top_q.first && top_q.second == m)
            break;
        
        ans++;
        pq.pop();
    }

    cout << ans + 1 << endl;
}

int     main()
{
    int t;
    scanf("%d", &t);

    for (int i = 0; i < t; i++)
        solve();

    return 0;
}