Codeforces1061C Multiplicity

태그:DP

제목.


제목 전송문

제목의 뜻


시퀀스 a i a 제시iai, 몇 개의 a의 하위 서열 b가 만족하는지 물어보기, 임의의 bibi bi​, b [ i ] m o d    i = 0 b[i]\mod i=0 b[i]modi=0

분석하다.


단순 DP
f [ i ] [ j ] = f [ i − 1 ] [ j − 1 ] + f [ i − 1 ] [ j ] ( a [ i ] m o d    j = 0 ) f[i][j]=f[i-1][j-1]+f[i-1][j] (a[i]\mod j=0) f[i][j]=f[i−1][j−1]+f[i−1][j](a[i]modj=0)
f [ i ] [ j ] = f [ i − 1 ] [ j ] ( a [ i ] m o d    j ̸ = 0 ) f[i][j]=f[i-1][j] (a[i]\mod jot= 0) f[i][j]=f[i−1][j](a[i]modj̸​=0)
이렇게 하면 분명히 MLE가 돼요.
나중에 최적화 고려
모든 i i i의 경우 j가 a[i]인 인수에서만 업데이트됩니다
같은 층에서 서로 영향을 줄 수 없기 때문에 큰 것부터 작은 것까지 처리해야 한다

code

#include
#include
#include
#include
#include
#include
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#define mem(x,num) memset(x,num,sizeof x)
#define reg(x) for(int i=last[x];i;i=e[i].next)
using namespace std;
inline ll read(){
    ll f=1,x=0;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
//******head by yjjr******
const int maxn=1e6+6,mod=1e9+7;
int f[maxn],cnt,n,v[maxn];
int main(){
    n=read();
    f[0]=1;
    rep(i,1,n){
        int x=read(),m=sqrt(x);cnt=0;
        rep(j,1,m)
            if(x%j==0){
                v[++cnt]=j;
                if(j*j!=x)v[++cnt]=x/j;
            }
        sort(v+1,v+1+cnt);
        dep(j,cnt,1)f[v[j]]=(f[v[j]]+f[v[j]-1])%mod;
    }
    ll ans=0;
    rep(i,1,n)ans=(ans+f[i])%mod;
    cout<<ans<<endl;
    return 0;
}

좋은 웹페이지 즐겨찾기