HDU5400 Arithmetic Sequence

解题思路：这题看懂题目是很关键的，这个区间是等差数列，且公差为d1或d2,

　　　　　特别注意单个数字也为等差数列。每次求出等差数列序列长度，然后

　　　　  求出对应这种长度对应有多少种组合方式，累加起来就是结果。

　　　　  注意要用long long,还有注意特判数据，如 5 -1 -1 ，5 4 3 2 1；

　　　　 5 1 1， 1 2 3 4 5 ; 5 1 1, 1 1 1 1 1等。

#include<cstdio>
int main()
{
    int A[100005], n, d1, d2;
    long long  sum, cnt; //注意这里要用long long 否则会WA
    //int sum, cnt;
    while(~scanf("%d %d %d", &n, &d1, &d2))
    {
        sum = cnt = 0;
        for(int i = 0; i < n; i++) scanf("%d", &A[i]);
        for(int i = 1; i < n; i++)
        {
            if(A[i]-A[i-1] ==  d1)
            {
                while(A[i]-A[i-1] == d1)
                {
                    i ++;
                    cnt ++;
                    if(i == n) break;
                }
                if(i == n) break; //i为n时要及时跳出，其它地方同理。
                while(A[i]-A[i-1] == d2)
                {
                    i ++;
                    cnt ++;
                    if(i == n) break;
                }
                sum += (cnt+1)*cnt/2; //满足条件的序列长度为cnt+1时，共有(cnt+1)*cnt/2种组合方式
                cnt = 0; //cnt重新初始化
                i --; //一定要回退一步，画画就知道了。
                continue;
            }
            if(i == n) break;
            if(A[i]-A[i-1] == d2)
            {
                while(A[i]-A[i-1] == d2)
                {
                    i ++;
                    cnt ++;
                    if(i == n) break;
                }
                sum += (cnt+1)*cnt/2;
                cnt = 0;
                i --;
            }
        }
        sum += (cnt+1)*cnt/2; //这步不能少
        printf("%I64d
", sum+n); //一定要加上这个n，刚开始我加的是5，WA了一发
       // printf("%d
", sum+n);
    }
    return 0;
}