HDU 4961 基础维护题

　　先附上题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4961

　　题意是给你一串数， 对于其中一个位置的数， 找出这两个数， 第一个数是这个数前面距离此数最近且为这个数的倍数的数， 第二个数是找出后面距离此数最近且是这个数的倍数的表， 对于每个位置的数， 将这个位置的两个数之积求出来再求和即可， 代码如下：

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>

using namespace std;
const int maxn = 100000 + 100;
vector<int> bei[maxn];  //存储一个数的倍数
int n;
int a[maxn];
int f[maxn], g[maxn];
int id[maxn];

int main() {
    //for(int i=2; i<=100000; i++) bei[1].push_back(i);
    for(int i=2; i<=100000; i++)
        for(int j=i; j<=100000; j+=i)
            bei[i].push_back(j);
    while(scanf("%d", &n) == 1 && n){
        for(int i=0; i<n; i++)
            scanf("%d", &a[i]);
        memset(id, -1, sizeof(id));
        for(int i=0; i<n; i++){
            int nownum = a[i];
            if(nownum == 1) {
                if(i-1<0) f[i] = a[i];
                else f[i] = a[i-1];
                continue;
            }
            int idx = -1;
            for(int j=0; j<bei[nownum].size(); j++){
                int tp = id[bei[nownum][j]];
                if(tp!=-1) idx = max(idx, tp);
            }
            if(idx!=-1) f[i] = a[idx];
            else f[i] = a[i];
            id[nownum] = i;
        }
        memset(id, -1, sizeof(id));
        for(int i=n-1; i>=0; i--){
            int nownum = a[i];
            if(nownum == 1) {
                if(i+1>=n) g[i] = a[i];
                else g[i] = a[i+1];
                continue;
            }
            int idx = 0x3f3f3f3f;
            for(int j=0; j<bei[nownum].size(); j++){
                int tp = id[bei[nownum][j]];
                if(tp!=-1) idx = min(idx, tp);
            }
            if(idx!=0x3f3f3f3f) g[i] = a[idx];
            else g[i] = a[i];
            id[nownum] = i;
        }
//        for(int i=0; i<n; i++) printf("%d ", a[f[i]]); printf("
");
//        for(int i=0; i<n; i++) printf("%d ", a[g[i]]); printf("
");
        long long res = 0;
        for(int i=0; i<n; i++){
            res = res + (long long)f[i]*(long long)g[i];
        }
        printf("%lld
", res);
    }
    return 0;
}