Submission #2114490


Source Code Expand

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

typedef long long i64;

i64 MOD = 1e9 + 7;

i64 mod(i64 num)
{
	if(num < 0)
	{
		num += MOD;
	}
	num = num % MOD;
	return num;
}

const i64 MAX = 2000000;

i64 dp[MAX + 100];

int n;

int main()
{
	cin >> n;

	dp[1] = 1;

	i64 sq = mod((n - 1) * (n - 1));
	i64 result = 0;

	for(int i = 1;i <= n;i++)
	{
		dp[i] = mod(dp[i] + dp[i - 1]);

		cout << i << "=" << dp[i] << endl;
		if(i == n)
		{
			result = mod(result + dp[i] * (n));
		}

		else 
		{
			dp[i + 3] = mod(dp[i + 3] + dp[i]);
			dp[i + n + 2] = mod(dp[i + n + 2] - dp[i]);

			int l = max(i + 3,n + 1);
			i64 len = i + n + 2 - l;

			result =  mod(result + len * dp[i]);

			result = mod(result +  dp[i] * sq);
		}
		//cout << i << "-----" << endl;

		//cout <<"----" << endl;
	}






	cout << result << endl;
	return 0;
}

Submission Info

Submission Time
Task F - Infinite Sequence
User niuez
Language C++14 (GCC 5.4.1)
Score 0
Code Size 937 Byte
Status WA
Exec Time 1833 ms
Memory 32256 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 1000
Status
WA × 2
WA × 22
Set Name Test Cases
Sample 0_000.txt, 0_001.txt
All 0_000.txt, 0_001.txt, max_1000000.txt, max_999745.txt, max_999880.txt, max_999999.txt, min_1.txt, rnd_14.txt, rnd_22.txt, rnd_25002.txt, rnd_2956.txt, rnd_3.txt, rnd_380467.txt, rnd_407774.txt, rnd_52228.txt, rnd_68.txt, rnd_804783.txt, rnd_85984.txt, rnd_894324.txt, rnd_93.txt, rnd_963981.txt, rnd_968416.txt
Case Name Status Exec Time Memory
0_000.txt WA 1 ms 256 KB
0_001.txt WA 1122 ms 21120 KB
max_1000000.txt WA 1833 ms 32256 KB
max_999745.txt WA 1726 ms 32256 KB
max_999880.txt WA 1716 ms 32256 KB
max_999999.txt WA 1765 ms 32256 KB
min_1.txt WA 1 ms 256 KB
rnd_14.txt WA 1 ms 256 KB
rnd_22.txt WA 1 ms 256 KB
rnd_25002.txt WA 45 ms 1024 KB
rnd_2956.txt WA 6 ms 384 KB
rnd_3.txt WA 1 ms 256 KB
rnd_380467.txt WA 661 ms 12544 KB
rnd_407774.txt WA 708 ms 15104 KB
rnd_52228.txt WA 94 ms 1792 KB
rnd_68.txt WA 1 ms 256 KB
rnd_804783.txt WA 1377 ms 27776 KB
rnd_85984.txt WA 149 ms 2944 KB
rnd_894324.txt WA 1546 ms 29184 KB
rnd_93.txt WA 1 ms 256 KB
rnd_963981.txt WA 1651 ms 31104 KB
rnd_968416.txt WA 1665 ms 31232 KB