Codeforces 777E - Hanoi Factory(贪心+栈)

题目链接:http://codeforces.com/problemset/problem/777/E

题意:有n个环给你内环半径、外环半径和高度,叠这些环还要满足以下要求:

    ①:下面的环的外径要>=上面的环

    ②:环不能掉下去,所以下面的环的内径要<上面的环的外径

    ③:叠出最大高度

思路:贪心思想,先把这些环按外径从大到小,内径从大到小排好序,内径从大到小是为了尽可能让后面的环能放上去。接下来就可以用栈模拟了,放1~n个环的时每次都要判断,如果当前最顶上的环的内径大于我要放的环的内径也就是放不上去,那就把顶上的环出栈,一直循环直到能放上去为止,再把当前环入栈。然后每次都比较一下高度和,得出最大高度。为什么可以用栈写?因为我们已经知道前面的环叠不上后面的环肯定也叠不上,因为后面的环外径肯定<=前面的环,那把栈当成已经建好的塔,如果环放不上去,为了后面可以继续放,那可以拆了一部分重建。

  这是开始没用栈的无脑代码从头到尾遍历了一遍,直接超时,心塞- -、、:

  

#include<iostream>
#include<algorithm>
#include<string>
using namespace std;
const int N=1e5+;
typedef long long ll;
struct node{
int in;
int out;
int h;
}a[N];
ll dp[N]; bool cmp(node a,node b){
return a.out==b.out?a.in>b.in:a.out>b.out;
} int main(){
ll n,ans=;
cin>>n;
for(int i=;i<=n;i++){
cin>>a[i].in>>a[i].out>>a[i].h;
if(a[i].h>ans)
ans=a[i].h;
}
sort(a+,a++n,cmp);
// cout<<endl;
// for(int i=1;i<=n;i++){
// cout<<a[i].in<<" "<<a[i].out<<" "<<a[i].h<<endl;
// }
for(int i=;i<=n;i++){
dp[i]=a[i].h;
for(int j=;j<i;j++){
if(a[i].out>=a[j].in)
dp[i]=max(dp[i],dp[j]+a[i].h);
}
ans=max(ans,dp[i]);
}
cout<<ans<<endl;
}

这是用栈写的:

#include<iostream>
#include<algorithm>
#include<stack>
using namespace std;
const int N=1e5+;
typedef long long ll;
struct node{
int in;
int out;
int h;
}a[N];
ll dp[N]; bool cmp(node a,node b){
return a.out==b.out?a.in>b.in:a.out>b.out;
}
int main(){
ll n;
stack<node>s;
cin>>n;
for(int i=;i<=n;i++){
cin>>a[i].in>>a[i].out>>a[i].h;
}
sort(a+,a++n,cmp);
// cout<<endl;
// for(int i=1;i<=n;i++){
// cout<<a[i].in<<" "<<a[i].out<<" "<<a[i].h<<endl;
// }
ll sum=,ans=;
for(int i=;i<=n;i++){
while(!s.empty()&&a[i].out<=s.top().in){//判断是否能叠上去,不能的话就把最顶上的环拿下来
sum-=s.top().h;
s.pop();
}
sum+=a[i].h;
s.push(a[i]);
ans=max(ans,sum); //比较最大高度
}
cout<<ans<<endl;
}
上一篇:codeforces-777E Hanoi Factory (栈+贪心)


下一篇:递归转手工栈处理的一般式[C语言]