問題
給定一個5個元素的數組(範圍0-13,對應撲克牌的),其中0可以代表任何數字,求這5個數字能否組成一個順子(即連續的5個數字,如2,0,4,5,0可以表示2,3,4,5,6)。
來源:牛客網
思路:模擬
隨便排個序,統計0的個數,模擬把0填入“空缺”裏,如果0不夠表示非法!同時注意有重複的數字也表示不可能構成順子!
#include <cstdio>
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
class Solution {
public:
bool IsContinuous(vector<int> nums) {
if (nums.size() != 5)
return false;
sort(nums.begin(), nums.end());
int zeroCount = 0, i = -1, len = nums.size();
while (++i < len && nums[i] == 0)
++zeroCount;
// 開始模擬填0,注意i是從第2個非零的數字開始的!
while (++i < len) {
if (nums[i] == nums[i-1])
return false;
zeroCount -= nums[i] - nums[i-1] - 1;
if (zeroCount < 0)
return false;
}
return true;
}
};
int main() {
int arr[] = {0, 3, 1, 6, 4};
vector<int> nums;
for (int i = 0; i < 5; ++i)
nums.push_back(arr[i]);
cout << Solution().IsContinuous(nums);
return 0;
}