MYF

HDU 5328 Problem Killer

Posted on | In | | Visitors

题目链接

HDU 5328
小模拟

题目分析

题目大意

给定n个数字形成的数列，这个数列中存在一段连续的等差或等比子区间，问最长的子区间长度。

解析

三个三个的往后扫即可。

Tips

  • 等比数列时相乘会超过int的范围，所以需要用long long来存储。
  • n=1或n=2的时候需要特判，因为任意一个数字or两个数字都是等差数列

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
#include <set>
#include <map>
#include <stack>
#include <cmath>
#include <queue>
#include <cstdio>
#include <string>
#include <vector>
#include <iomanip>
#include <cstring>
#include <iostream>
#include <algorithm>
#define Memset(a,val) memset(a,val,sizeof(a))
#define PI acos(-1)
#define rt(n)       (i == n ? '\n' : ' ')
#define hi         printf("Hi----------\n")
#define IN freopen("input.txt","r",stdin)
#define OUT freopen("output.txt","w",stdout)
#define debug(x) cout<<"Debug : ---"<<x<<"---"<<endl;
#pragma comment(linker, "/STACK:1024000000,1024000000")
typedef long long ll;
using namespace std;
const int maxn=1000000+5;
const int mod=1000000007;
const int INF=0x3f3f3f3f;
const double eps=1e-8;
ll a[maxn];
void work(){
    int n;
    scanf("%d",&n);
    for (int i=1; i<=n; i++) {
        scanf("%lld",&a[i]);
    }
    if (n==1||n==2) {
        printf("%d\n",n);
        return;
    }
    int ans=2,id=1;
    while (id!=n-1) {
        int mx=2;
        ll aa=a[id],bb=a[id+1],cc=a[id+2];
        if (aa+cc==bb*2) {
            while (aa+cc==bb*2) {
                mx++;
                id++;
                if (id==n-1)
                    break;
                aa=a[id],bb=a[id+1],cc=a[id+2];
            }
            ans=max(ans, mx);
        }
        else if (aa*cc==bb*bb){
            while (aa*cc==bb*bb) {
                mx++;
                id++;
                if (id==n-1)
                    break;
                aa=a[id],bb=a[id+1],cc=a[id+2];
            }
            ans=max(ans, mx);
        }
        else
            id++;
    }
    printf("%d\n",ans);
    return;
}
int main(){
    int t;
    cin>>t;
    while (t--) {
        work();
    }
}