【Hihocoder - offer编程练习赛86 - C】跳数游戏(STLset或map预处理,dp)
題干:
時間限制:10000ms
單點時限:1000ms
內存限制:256MB
描述
給定一個長度為N的數組A1, A2, ... AN,請你判斷其中有幾個元素Ai按如下跳躍規則能跳到最后一個元素AN。 ?
假設你當前位于Ai,跳躍的規則是:
如果這一步是第奇數次跳躍(從1開始計數),可以跳到Ai之后(Ai+1?.. AN)比Ai大的最小的元素;
如果這一步是第偶數次跳躍,可以跳到Ai之后(Ai+1?.. AN)比Ai小的最大的元素;
如果有多個滿足條件的元素,則跳到其中下標最小的元素。 ?
如果沒有滿足條件的元素,則在當前的Ai停下來。 ?
例如對于A = [3, 2, 4, 1, 5],如果從3開始,第一步從3跳到4,第二步從4跳到1,第三步從1跳到5。
輸入
第一行包含一個整數N。 ?
第二行包含N個整數A1, A2, ... AN。 ?
對于30%的數據,1 <= N <= 100 ?
對于60%的數據,1 <= N <= 1000 ?
對于100%的數據,1 <= N <= 100000 ?1 <= Ai?<= 1000000
輸出
一個整數代表答案
樣例輸入
5 3 4 1 2 5樣例輸出
4解題報告:
? ?先用set倒著掃一遍數組,順便預處理出距離最近的比他大的最小的數的下標,和距離最近的比他小的最大的數的下標,這一步可以用單調棧O(n)實現,但是因為時間允許于是set好寫一些。然后dp代表從編號i開始的第偶數/奇數步,能不能到達最后。
AC代碼:
#include<cstdio> #include<iostream> #include<algorithm> #include<queue> #include<map> #include<vector> #include<set> #include<string> #include<cmath> #include<cstring> #include<cctype> #define ll long long #define pb push_back #define pm make_pair using namespace std; const int MAX = 2e6 + 5; set<int> ss; bool dp[MAX][2]; int big[MAX],small[MAX],pos[MAX],a[MAX];//預處理出后面第一個比他大的和第一個比他小的 int main() {int n;cin>>n;for(int i = 1; i<=n; i++) scanf("%d",a+i);for(int i = n; i>=1; i--) {if(ss.count(a[i])) pos[a[i]] = i;else ss.insert(a[i]),pos[a[i]] = i;auto it = ss.upper_bound(a[i]);if(it != ss.end()) big[i] = pos[*it];else big[i] = -1;it = ss.lower_bound(a[i]);if(it == ss.begin()) small[i] = -1;else {--it;small[i] = pos[*it];}} // for(int i = 1; i<=n; i++ ) { // printf("i: %d : %d %d\n",i,big[i],small[i]); // }dp[n][0] = 1,dp[n][1] = 1;for(int i = n-1; i>=1; i--) {if(big[i] != -1) dp[i][1] = dp[big[i]][0];if(small[i] != -1) dp[i][0] = dp[small[i]][1];}int ans = 0;for(int i = 1; i<=n; i++) {if(dp[i][1]) ans++;}printf("%d\n",ans);return 0 ;}注意因為題中的大于和小于都是嚴格大于和嚴格小于,所以判斷大于必須用upper,否則可能set中已經有一個a[i]了,,你再lower就肯定不對了。
同樣的,也可以用map實現:
#include <bits/stdc++.h> using namespace std; const int N = 1e5+55; int dp[N][2]; int a[N]; int main() {int n;cin>>n;for(int i=1;i<=n;++i)cin>>a[i];map<int,int> mp;map<int,int> ::iterator it;mp[a[n]] = n;dp[n][1] = 1;dp[n][0] = 1;for(int i=n-1;i>0;--i){it = mp.lower_bound(a[i]);if(it==mp.end()||it==mp.begin()) dp[i][0] = 0;else dp[i][0] = dp[(--it)->second][1];it = mp.upper_bound(a[i]);if(it==mp.end()) dp[i][1] = 0;else dp[i][1] = dp[it->second][0];mp[a[i]]=i;}int ans = 0;for(int i=1;i<=n;++i){if(dp[i][1]) ans++;}cout<<ans;return 0; }?
總結
以上是生活随笔為你收集整理的【Hihocoder - offer编程练习赛86 - C】跳数游戏(STLset或map预处理,dp)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 六大行集体清理一类高息存款,高息存款难觅
- 下一篇: 【HDU - 5961】传递(图,思维,