Codeforces 777E Hanoi Factory
题目链接
http://codeforces.com/contest/777/problem/E
题目大意
有一个类似于汉诺塔的游戏,只是每个圆盘中间都是空心的,一个环 $i$ 能放在另一个环 $j$ 上当且仅当它的外半径小于 $j$ 环的外半径且外半径大于 $j$ 环的内半径,否则它将掉下去。现在已知每个环的高度,需要你求出最高能摆多高。
题目分析
这题注意到一个性质,如果对于一个环 $i$ ,它因为外半径过小无法放到圆环 $j$ 上,那么所有外半径小于等于 $i$ 的均无法放到圆环 $j$ 上。那么可以贪心求解。先将所有的圆环按外半径大小从大到小排序,如果外半径相同则按内半径大小从大到小排序,用栈来维护当前已经搭好的汉诺塔部分,每放进一个新的圆盘,将栈顶所有无法放它的圆盘全部弹出,记录高度,取最大值即可
时间复杂度: $O(N \times \log_{2}{N})$
代码
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
using namespace std;
int n;
struct ex {
    long long a;
    long long b;
    long long h;
};
struct sta {
    long long h[100005];
    long long a[100005];
    int p;
    
    long long topa(){
        return a[p];
    }
    
    long long toph(){
        return h[p];
    }
    
    int push(long long t, long long ht){
        p++;
        a[p] = t;
        h[p] = h[p - 1] + ht;
        return 0;
    }
    
    int pop(){
        p--;
        return 0;
    }
    
    sta(){p = 0;h[0] = 0;a[0] = 0;};
};
ex ring[100005];
sta s;
long long maxh = 0;
bool cmp(ex a, ex b){
    if(a.b != b.b){
        return a.b > b.b;
    }else{
        return a.a > b.a;
    }
}
int main(){
    std::ios::sync_with_stdio(false);
    
    int i, j;
    
    cin >> n;
    
    for(i=1;i<=n;i++){
        cin >> ring[i].a;
        cin >> ring[i].b;
        cin >> ring[i].h;
    }
    
    sort(ring + 1, ring + n + 1, cmp);
    
    for(i=1;i<=n;i++){
        while(ring[i].b <= s.topa()){
            s.pop();
        }
        s.push(ring[i].a, ring[i].h);
        maxh = max(maxh, s.toph());
    }
    
    cout << maxh << endl;
    
    return 0;
}