HDU 5143 NPY and arithmetic progression 思维+枚举

HDU 5143
题意:1,2,3,4的个数分别为a,b,c,d 问用这些数是否能组成若干个长度>=3的等差数列.
a,b,c,d<=1e9.
首先,AP(等差数列)只有6种,111(...),222(..),333(...),123,234,1234
关键点:3个123 可以看作3个1,3个2,3个3. 234,1234也类似.

所以最后结果只有 常数列+ 0~2个123,0~2个234,0~2个1234 27种情况 直接暴力枚举即可.

#include <bits/stdc++.h>
using namespace std;
typedef pair<int,int> ii;
typedef long long ll;
const int N=2e5+5;
int a[5];
bool check(int x,int y,int z)
{
    int one=a[1]-(x+z);
    int two=a[2]-(x+y+z);
    int three=a[3]-(x+y+z);
    int four=a[4]-(y+z);
    return ((one>=3||one==0) && (two>=3||two==0)&& (three==0||three>=3)&&(four==0||four>=3));
}
int main()
{
    int T,n=4;
    cin>>T;
    while(T--)
    {
        bool flag=false;
        for(int i=1;i<=n;i++)
            cin>>a[i];
        for(int i=0;i<=2;i++)
            for(int j=0;j<=2;j++)
                for(int k=0;k<=2;k++)
                    if(check(i,j,k))
                        flag=true;
        puts(flag?"Yes":"No");
    }
    return 0;
}