結果

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

int N;
int A[100000];
modint dp[100000];
modint C[101][100]; 

int main(){
	scanf("%d",&N);
	rep(i,N) scanf("%d",&A[i]);
	dp[0]=1;
	int m[101]={};
	rep(i,N){
		for(int d=1; d<=100; 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]<=100){ 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<=100; d++){ m[d]++; if(m[d]==d) m[d]=0; }
	}
	printf("%u\n",dp[N-1].v);
	return 0;
}
0