알고리즘 > 자료구조A > 탑(1809)

Question

문제 링크

Solving

Code

Colored By Color Scripter™

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57

#include <stdio.h> #include <stack> #pragma warning(disable:4996)   using namespace std;   typedef struct _fac {     int idx;     int val; }fac;   int n, cnt; fac building[500005]; stack<fac> box; int answer[500005];   int main() {     FILE *fin = fopen("input.txt", "rt");     FILE *fout = fopen("output.txt", "wt");       fscanf(fin, "%d", &n);     cnt = n;     for (int i = 1; i <= n; i++)     {         fscanf(fin, "%d", &building[i].val);                 building[i].idx = i;     }       box.push(building[n]);     for (int i = n-1; i >= 1; i--)     {         while (1)         {             if (box.empty())             {                 box.push(building[i]);                 break;             }             else if (box.top().val < building[i].val)             {                 answer[box.top().idx] = building[i].idx;                 box.pop();             }             else if (box.top().val > building[i].val)             {                 box.push(building[i]);                 break;             }         }     }       for (int i = 1; i <= n; i++)         fprintf(fout, "%d ", answer[i]);     fprintf(fout, "\n"); }