leetcode----967. 連續差相同的數字

https://leetcode-cn.com/problems/numbers-with-same-consecutive-differences/

返回所有長度爲 N 且滿足其每兩個連續位上的數字之間的差的絕對值爲 K 的非負整數

請注意,除了數字 0 本身之外,答案中的每個數字都不能有前導零。例如,01 因爲有一個前導零,所以是無效的;但 0 是有效的。

你可以按任何順序返回答案。

示例 1:

輸入:N = 3, K = 7
輸出:[181,292,707,818,929]
解釋:注意,070 不是一個有效的數字,因爲它有前導零。

示例 2:

輸入:N = 2, K = 1
輸出:[10,12,21,23,32,34,43,45,54,56,65,67,76,78,87,89,98]

提示:

  1. 1 <= N <= 9
  2. 0 <= K <= 9

1對於邊界,一定要考慮周全

2.這裏dfs裏面是先賦值,再看看有沒有達到最終條件。

class Solution {
	public:
		vector<int> numsSameConsecDiff(int N, int K) {
			vector< vector<int> > vt(10);
			for(int i=0; i<10; i++) {
				if(i+K<10&&K!=0)
					vt[i].push_back(i+K);
				if(i-K>=0&&K!=0)
					vt[i].push_back(i-K);

				if(K==0)
					vt[i].push_back(i);
			}

			for(int i=0; i<10; i++) {
				dfs(i,0,N,vt);
			}
			return res;

		}
		void dfs(int val,int ilevel,int N,vector< vector<int> > vt) {
			a[ilevel]=val;
			ilevel++;
			if(ilevel==N) {
				int sum=0;
				if(N>1&&a[0]==0) return;
				for(int i=0; i<N; i++) {
					sum=sum*10+a[i];
				}
				res.push_back(sum);
				return;
			}

			for(int i=0; i<vt[val].size(); i++) {
				dfs(vt[val][i],ilevel,N,vt);
			}
		}
		int a[10];

		vector<int> res;
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章