HDU 5143 DFS

Posted 2020-10-08 ( m Lweleth)

分别给出1,2,3,4   a, b, c,d个

问能否组成数个长度不小于3的等差数列。

首先数量存在大于3的可以直接拿掉，那么可以先判是否都是0或大于3的

然后直接DFS就行了，但是还是要注意先判合法能否进入下层递归来减少内存消耗。

 

/** @Date    : 2017-09-27 15:08:23
  * @FileName: HDU 5143 DFS.cpp
  * @Platform: Windows
  * @Author  : Lweleth ([email protected])
  * @Link    : https://github.com/
  * @Version : $Id$
  */
#include <bits/stdc++.h>
#define LL long long
#define PII pair<int ,int>
#define MP(x, y) make_pair((x),(y))
#define fi first
#define se second
#define PB(x) push_back((x))
#define MMG(x) memset((x), -1,sizeof(x))
#define MMF(x) memset((x),0,sizeof(x))
#define MMI(x) memset((x), INF, sizeof(x))
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;

const int INF = 0x3f3f3f3f;
const int N = 1e5+20;
const double eps = 1e-8;

int ans = 0;
bool dfs(int a, int b, int c, int d)
{
	if(ans)
		return 1;
	if(!a && !b && !c && !d)
		return 1;
	else if((a==0 || a > 2) && (b==0 || b > 2) && (c==0 || c > 2) && (d==0 || d > 2))
		return 1;
	else if(a < 0 || b < 0 || c < 0 || d < 0)
		return 0;
	if(!ans)
		ans |= dfs(a - 1, b - 1, c - 1, d - 1);
	if(!ans)
		ans |= dfs(a - 1, b - 1, c - 1, d);
	if(!ans)
		ans |= dfs(a, b - 1, c - 1, d - 1);
	return ans;
}

int main()
{
	/*int size = 256 << 20; // 256MB  
    char *p = (char*)malloc(size) + size;  
    __asm__("movl %0, %%esp\n" :: "r"(p));*/
	int T;
	cin >> T;
	while(T--)
	{
		int a, b, c, d;
		scanf("%d%d%d%d", &a, &b, &c, &d);
		ans = 0;
		printf("%s\n", dfs(a, b, c, d)?"Yes":"No");
	}
    return 0;
}