hdu 4961 Boring Sum

题目链接:点击打开链接


题意很好理解,实际上转换一下,b[i]代表的就是在a[i]前面的第一个倍数,如果没有就等于a[i],,c[i]就是a[i]后面的第一个倍数。

这样先预处理出1到100000所有数的约数存入p[ ],用k[ ]表示1到100000中每个数紧挨着的倍数。先从前往后扫扫到每个数就根据这个数的约数更新k[ ]得到b[ ],再从后往前得到c[ ]。


代码:

#include <cstdio>
#include <iostream>
#include <vector>
#define MAX 100010
#include <cstring>
using namespace std;

int a[MAX];
int b[MAX];
int c[MAX];
int k[MAX];
bool vis[MAX];

vector <int > p[MAX];
int main(){
    int n;
    for(int i=1;i<=100000;i++){
        for(int j=i;j<=100000;j+=i){
            p[j].push_back(i);
        }
    }
    while(~scanf("%d",&n),n){
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
        }
        for(int  i=1;i<=n;i++){
            if(!vis[a[i]]) b[i]=a[i];
            else b[i]=k[a[i]];
            for(int j=0;j<p[a[i]].size();j++){
                k[p[a[i]][j]]=a[i];
                vis[p[a[i]][j]]=1;
            }
        }
        memset(vis,0,sizeof(vis));
        for(int  i=n;i>=1;i--){
            if(!vis[a[i]]) c[i]=a[i];
            else c[i]=k[a[i]];
            for(int j=0;j<p[a[i]].size();j++){
                k[p[a[i]][j]]=a[i];
                vis[p[a[i]][j]]=1;
            }
        }
        long long res=0;
        for(int i=1;i<=n;i++){
            res+=(long long)b[i]*(long long)c[i];
        }
        printf("%I64d\n",res);
    }
    return 0;
}





你可能感兴趣的:(hdu 4961 Boring Sum)