洛谷P2866《[USACO06NOV]糟糕的一天Bad Hair Day》
阿新 • • 發佈:2020-10-14
原更新時間:2018-10-06 21:28:46
有點難想的單調棧模板題
暫不提供題面,請自行到洛谷檢視題面。
Input / Output 格式 & 樣例
輸入格式
第一行:一個數N表示奶牛的數量。
第2到N+1行:第i+1行包含一個整數表示第i頭奶牛的高。
輸出格式
第一行:一個整數,即c1到cN的和
輸入樣例
6
10
3
7
4
12
2
輸出樣例
5
解題思路
這就是一個單調棧的模板
迴圈讀入,每次push讀入的數進一個單調棧並維護這個棧的單調性,最後答案累加棧的大小-1即可(顯然題意說明奶牛是看不見自己的髮型的,要把自己減去)
以上操作的推導過程:
- 我們對於當前讀進去的奶牛的高度,計算棧中還有多少比它矮的,把它們pop出來(維護單調性)
- 這個過程其實就是在計算當前的奶牛能被多少奶牛看見
- 接下來累計答案
- 最後將當前奶牛的高度push進去
程式碼實現
/* -- Basic Headers -- */ #include <iostream> #include <cstdio> #include <cstring> #include <cctype> #include <algorithm> /* -- STL Iterators -- */ #include <vector> #include <string> #include <stack> #include <queue> /* -- External Headers -- */ /* -- Defined Functions -- */ #define For(a,x,y) for (int a = x; a <= y; ++a) #define Forw(a,x,y) for (int a = x; a < y; ++a) #define Bak(a,y,x) for (int a = y; a >= x; --a) /* -- Defined Words -- */ using namespace std; namespace FastIO { void DEBUG(char comment[], int x) { cerr << comment << x << endl; } inline int getint() { int s = 0, x = 1; char ch = getchar(); while (!isdigit(ch)) { if (ch == '-') x = -1; ch = getchar(); } while (isdigit(ch)) { s = s * 10 + ch - '0'; ch = getchar(); } return s * x; } inline void __basic_putint(int x) { if (x < 0) { x = -x; putchar('-'); } if (x >= 10) __basic_putint(x / 10); putchar(x % 10 + '0'); } inline void putint(int x, char external) { __basic_putint(x); putchar(external); } } namespace Solution { const int MAXN = 80000 + 10; struct Stack { int seq[MAXN]; int tail; Stack() { memset(seq, 0, sizeof(seq)); tail = 0; } void Pop() { tail--; } int Top() { return seq[tail]; } bool isEmpty() { return tail == 0; } void Push(int x) { while (!isEmpty() && Top() <= x) Pop(); seq[++tail] = x; } int Size() { return tail; } int __tail_location() { return tail; } } stk; // 手寫棧無所畏懼 int n; } int main(int argc, char *const argv[]) { #ifdef HANDWER_FILE freopen("testdata.in", "r", stdin); freopen("testdata.out", "w", stdout); #endif using namespace Solution; using namespace FastIO; n = getint(); long long int ans = 0; For (i, 1, n) { int x = getint(); stk.Push(x); ans += stk.Size() - 1; } cout << ans << endl; return 0; }