結果

問題 No.1331 Moving Penguin
コンテスト
ユーザー Nachia
提出日時 2021-01-08 22:43:03
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 162 ms / 1,500 ms
コード長 818 bytes
コンパイル時間 3,426 ms
コンパイル使用メモリ 193,308 KB
最終ジャッジ日時 2025-01-17 13:10:47
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 49
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:23:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   23 |         scanf("%d",&N);
      |         ~~~~~^~~~~~~~~
main.cpp:24:23: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   24 |         rep(i,N) scanf("%d",&A[i]);
      |                  ~~~~~^~~~~~~~~~~~

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using LL=long long;
using ULL=unsigned long long;
#define rep(i,n) for(int i=0; i<(n); i++)

const unsigned int M=1000000007;

struct modint{
	unsigned int v=0;
	modint(unsigned int a=0){ v=a; }
	void operator+=(modint r){ v+=r.v; if(v>=M) v-=M; }
};

const int sqZ=256;

int N;
int A[100000];
modint dp[100001];
modint C[sqZ+1][sqZ]; 

int main(){
	scanf("%d",&N);
	rep(i,N) scanf("%d",&A[i]);
	dp[0]=1;
	int m[sqZ+1]={};
	rep(i,N){
		for(int d=1; d<=sqZ; d++) dp[i]+=C[d][m[d]];
		if(i==N-1) break;
		if(A[i]==1){ C[1][0]+=dp[i]; }
		else if(A[i]<=sqZ){ dp[i+1]+=dp[i]; C[A[i]][m[A[i]]]+=dp[i]; }
		else{ dp[i+1]+=dp[i]; for(int j=i+A[i]; j<N; j+=A[i]) dp[j]+=dp[i]; }
		for(int d=1; d<=sqZ; d++){ m[d]++; if(m[d]==d) m[d]=0; }
	}
	printf("%u\n",dp[N-1].v);
	return 0;
}
0