HDU 5143 NPY and arithmetic progression（思维）

http://acm.hdu.edu.cn/showproblem.php?pid=5143

题意：

给定数字1,2,3,4.的个数每个数字能且仅能使用一次，组成多个或一个等差数列（长度大于等于3），问能否成功。

思路：

可以发现等差数列只有（123,234,1234和长度>=3的常数列），如果选择非常数数列（123,234,1234）数量大于等于3，

可以变为三个或4个常数列,例如（123,123,123）变为（111,222,333）。所以从0-2枚举选择非常数列的数量，再判断能否用常数列覆盖剩下的（如果数字长度正好为0或≤3就可以）。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<sstream>
#include<vector>
#include<stack>
#include<queue>
#include<cmath>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
const int INF = 0x3f3f3f3f;
const int maxn = 10000+5;

int a1,a2,a3,a4;

bool check(int a, int b, int c, int d)
{
    if((a>=3||a==0)&&(b>=3||b==0)&&(c>=3||c==0)&&(d>=3||d==0))  return true;
    return false;
}

int main()
{
    //freopen("in.txt","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d%d",&a1,&a2,&a3,&a4);
        bool flag = false;
        if(check(a1,a2,a3,a4))  flag=true;
        else
        for(int i=0;i<=2;i++)
        {
            for(int j=0;j<=2;j++)
            {
                for(int k=0;k<=2;k++)
                {
                    if(check(a1-i-k,a2-i-j-k,a3-i-j-k,a4-j-k))  flag=true;
                }
            }
        }
        if(flag)  puts("Yes");
        else puts("No");
    }
    return 0;
}