团体程序设计天梯赛-练习集 L2-012. 关于堆的判断

团体程序设计天梯赛-练习集
L2-012. 关于堆的判断
https://www.patest.cn/contests/gplt/L2-012
题的关键在理解上，到底是用push_heap 还是 make_heap要理解清楚。
最后就是字符串的处理了。

#include <iostream>
#include <queue>
#include <vector>
#include <sstream>
#include <algorithm>
#include <cstdio>
#include <functional>
using namespace std;

vector<int>heap;
int find(int a)
{
    return distance(heap.begin(), find(heap.begin(), heap.end(), a));
}
int main()
{
    int n, m;
    scanf("%d%d", &n, &m);
    while (n--)
    {
        int t;
        scanf("%d", &t);
        heap.push_back(t);
        make_heap(heap.begin(), heap.end(), greater<int>());
    }
    getchar();
    while (m--)
    {
        string q;
        getline(cin, q);
        if (q[q.length() - 1] == 't')
        {
            int a;
            stringstream ss(q);
            ss >> a;
            if (heap[0] == a) printf("T\n");
            else printf("F\n");
        }
        else if (q[q.length() - 1] == 's')
        {
            int a, b;
            string temp;
            stringstream ss(q);
            ss >> a >> temp >> b;
            int pos_a = find(a);
            int pos_b = find(b);
            pos_a++; pos_b++;
            if (pos_a / 2 == pos_b / 2)
            {
                printf("T\n");
            }
            else
            {
                printf("F\n");
            }
        }
        else
        {
            stringstream ss(q);
            int a, b;
            string temp;
            ss >> a >> temp >> temp;
            if (temp[0] == 't')
            {
                ss >> temp >> temp >> b;
                int pos_a = find(a);
                int pos_b = find(b);
                if (pos_b == pos_a * 2 + 1 || pos_b == pos_a * 2 + 2)
                {
                    printf("T\n");
                }
                else
                {
                    printf("F\n");
                }
            }
            else if (temp[0] == 'a')
            {
                ss >> temp >> temp >> b;
                int pos_a = find(a);
                int pos_b = find(b);
                if (pos_a == pos_b * 2 + 1 || pos_a == pos_b * 2 + 2)
                {
                    printf("T\n");
                }
                else
                {
                    printf("F\n");
                }
            }
        }
    }
    return 0;
}