https://atcoder.jp/contests/abc161/tasks/abc161_d
にある問題を解いていたのですが、再帰関数を用いた実装でエラーとなってしまいました。
エラーとなった実装は1番目のコードです。このコードの中のketa++というところを、少し変えると解決したのですが(2番目のコード)自分でも理由がよくわかりませんでした。原因がわかる方がいらっしゃればお聞きしたいです。
c++
1#include <bits/stdc++.h> 2using namespace std; 3typedef long long ll; 4typedef long double ld; 5typedef pair<int, int> P; 6const ll mod = 1000000007; 7const ll INF = 1e+14; 8#define rep(i,n) for(int i=0;i<n;i++) 9#define per(i,n) for(int i=n-1;i>=0;i--) 10#define Rep(i,sta,n) for(int i=sta;i<n;i++) 11#define rep1(i,n) for(int i=1;i<=n;i++) 12#define per1(i,n) for(int i=n;i>=1;i--) 13#define Rep1(i,sta,n) for(int i=sta;i<=n;i++) 14#define _GLIBCXX_DEBUG 15 16 17void dfs (ll keta , ll n , vector<ll> &a){ 18 if(keta==11){ 19 return; 20 } 21 22 a.push_back(n); 23 for(int i=-1;i<2;i++){ 24 int res=n%10; 25 if(res+i>=0&&res+i<=9) { 26 keta++; 27 dfs(keta,n*10+res+i,a); 28 } 29 } 30} 31 32int main(){ 33 int k; 34 cin>>k; 35 vector<ll> a; 36 rep1(i,9){ 37 dfs(1,i,a); 38 } 39 sort(a.begin(),a.end()); 40 cout<<a[k-1]<<endl; 41} 42
c++
1#include <bits/stdc++.h> 2using namespace std; 3typedef long long ll; 4typedef long double ld; 5typedef pair<int, int> P; 6const ll mod = 1000000007; 7const ll INF = 1e+14; 8#define rep(i,n) for(int i=0;i<n;i++) 9#define per(i,n) for(int i=n-1;i>=0;i--) 10#define Rep(i,sta,n) for(int i=sta;i<n;i++) 11#define rep1(i,n) for(int i=1;i<=n;i++) 12#define per1(i,n) for(int i=n;i>=1;i--) 13#define Rep1(i,sta,n) for(int i=sta;i<=n;i++) 14#define _GLIBCXX_DEBUG 15 16 17void dfs (ll keta , ll n , vector<ll> &a){ 18 if(keta==11){ 19 return; 20 } 21 22 a.push_back(n); 23 for(int i=-1;i<2;i++){ 24 int res=n%10; 25 if(res+i>=0&&res+i<=9) { 26 dfs(keta+1,n*10+res+i,a); 27 } 28 } 29} 30 31int main(){ 32 int k; 33 cin>>k; 34 vector<ll> a; 35 rep1(i,9){ 36 dfs(1,i,a); 37 } 38 sort(a.begin(),a.end()); 39 cout<<a[k-1]<<endl; 40} 41
回答1件
あなたの回答
tips
プレビュー