CSP 202104-4 校门外的树 DP

原题链接：CSP 202104-4 校门外的树
参考博客：csp 2021-04-4 校门外的树
第22次CSP认证 第4题 校门外的树（3种方法，非常详细）（类dp+数学）
学习学习······

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int MAX=1e3+10;
const int M=1e5+10;
const ll mod=1e9+7;

vector<int> q[M];//每个数的因子
ll f[MAX];//f[i]为到第i个障碍物的方案总数
ll a[MAX];//输入
int flag[M];//状态数组

int main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(0);
    std::cout.tie(0);
    int n;
    cin>>n;
    for(int i=0;i<n;i++)
        cin>>a[i];
    //计算每个数的因子
    for(int i=1;i<M;i++)
    {
        for(int j=2*i;j<M;j+=i)
        {
            q[j].push_back(i);
        }
    }
    f[0]=1;
    for(int i=1;i<n;i++)//每轮添加一个障碍物
    {
        memset(flag,0,sizeof(flag));//清空状态数组
        for(int j=i-1;j>=0;j--)
        {
            int d=a[i]-a[j];
            ll cnt=0;// ll 型，因为方案可能很大 
            //枚举距离d的所有因子，记录所有的方案
            for(int k=0;k<q[d].size();k++)
            {
                int tmp=q[d][k];
                //如果这个方案之前没有出现过，方案数+1
                if(flag[tmp]==0)
                {
                    cnt++;
                    //标记当前因子被使用过了
                    flag[tmp]=1;
                }
            }
            //下一轮如果按照d间隔种树就会撞上该障碍物，所以要手动标记
            flag[d]=1;
            //更新
            f[i]=(f[i]+f[j]*cnt)%mod;
        }
    }
    cout<<f[n-1];
    return 0;
}