【Hihocoder - offer编程练习赛86 - C】跳数游戏(STLset或map预处理,dp)
題干:
時(shí)間限制:10000ms
單點(diǎn)時(shí)限:1000ms
內(nèi)存限制:256MB
描述
給定一個(gè)長(zhǎng)度為N的數(shù)組A1, A2, ... AN,請(qǐng)你判斷其中有幾個(gè)元素Ai按如下跳躍規(guī)則能跳到最后一個(gè)元素AN。 ?
假設(shè)你當(dāng)前位于Ai,跳躍的規(guī)則是:
如果這一步是第奇數(shù)次跳躍(從1開(kāi)始計(jì)數(shù)),可以跳到Ai之后(Ai+1?.. AN)比Ai大的最小的元素;
如果這一步是第偶數(shù)次跳躍,可以跳到Ai之后(Ai+1?.. AN)比Ai小的最大的元素;
如果有多個(gè)滿足條件的元素,則跳到其中下標(biāo)最小的元素。 ?
如果沒(méi)有滿足條件的元素,則在當(dāng)前的Ai停下來(lái)。 ?
例如對(duì)于A = [3, 2, 4, 1, 5],如果從3開(kāi)始,第一步從3跳到4,第二步從4跳到1,第三步從1跳到5。
輸入
第一行包含一個(gè)整數(shù)N。 ?
第二行包含N個(gè)整數(shù)A1, A2, ... AN。 ?
對(duì)于30%的數(shù)據(jù),1 <= N <= 100 ?
對(duì)于60%的數(shù)據(jù),1 <= N <= 1000 ?
對(duì)于100%的數(shù)據(jù),1 <= N <= 100000 ?1 <= Ai?<= 1000000
輸出
一個(gè)整數(shù)代表答案
樣例輸入
5 3 4 1 2 5樣例輸出
4解題報(bào)告:
? ?先用set倒著掃一遍數(shù)組,順便預(yù)處理出距離最近的比他大的最小的數(shù)的下標(biāo),和距離最近的比他小的最大的數(shù)的下標(biāo),這一步可以用單調(diào)棧O(n)實(shí)現(xiàn),但是因?yàn)闀r(shí)間允許于是set好寫(xiě)一些。然后dp代表從編號(hào)i開(kāi)始的第偶數(shù)/奇數(shù)步,能不能到達(dá)最后。
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];//預(yù)處理出后面第一個(gè)比他大的和第一個(gè)比他小的 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 ;}注意因?yàn)轭}中的大于和小于都是嚴(yán)格大于和嚴(yán)格小于,所以判斷大于必須用upper,否則可能set中已經(jīng)有一個(gè)a[i]了,,你再lower就肯定不對(duì)了。
同樣的,也可以用map實(shí)現(xiàn):
#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; }?
總結(jié)
以上是生活随笔為你收集整理的【Hihocoder - offer编程练习赛86 - C】跳数游戏(STLset或map预处理,dp)的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: 六大行集体清理一类高息存款,高息存款难觅
- 下一篇: 【HDU - 5961】传递(图,思维,