Submission #1084114


Source Code Expand

//satyaki3794
#include <bits/stdc++.h>
#define ff first
#define ss second
#define pb push_back
#define MOD (1000000007LL)
#define LEFT(n) (2*(n))
#define RIGHT(n) (2*(n)+1)
 
using namespace std;
typedef long long ll;
typedef pair<int, int> ii;
typedef pair<int, ii> iii;

ll pwr(ll base, ll p, ll mod = MOD){
ll ans = 1;while(p){if(p&1)ans=(ans*base)%mod;base=(base*base)%mod;p/=2;}return ans;
}
 

ll gcd(ll a, ll b){
    if(b == 0)  return a;
    return gcd(b, a%b);
}



int n, k;
ll DP[2002][2002][2], ways[2002];


ll dp(int i, int left, bool ok){
	int used = n-i  - left;
	if(left < 0)	return 0;
	if(used > k)	return 0;
	if(i == 1 && used != k){
		ll ans = 0;
		if(left > 0 && ok)
			ans = dp(1, left-1, true);
// cout<<"dp "<<i<<" "<<left<<" "<<ok<<" returns "<<ans<<endl;
		return ans;
	}
	if(used == k || i == 1){
// cout<<"dp "<<i<<" "<<left<<" "<<ok<<" returns 1"<<endl;
		return 1;
	}
	ll &ans = DP[i][left][ok];
	if(ans != -1)	return ans;

	ans = dp(i-1, left, true) + dp(i-1, left+1, false);
	ans %= MOD;
	if(left > 0 && ok){
		ans += dp(i, left-1, true);
		ans %= MOD;
	}
// cout<<"dp "<<i<<" "<<left<<" "<<ok<<" returns "<<ans<<endl;
	return ans;
}


int main(){
 
    ios_base::sync_with_stdio(0);
    cin.tie(0);

    cin>>n>>k;

    ways[0] = ways[1] = 1;
    for(int i=2;i<=n;i++)
    	ways[i] = (ways[i-1] * 2) % MOD;

    memset(DP, -1, sizeof(DP));
    k--;
    ll ans = (dp(n, 0, false) * ways[n-k-1]) % MOD;
    cout<<ans;
    return 0;
}






Submission Info

Submission Time
Task F - Solitaire
User satyaki3794
Language C++14 (GCC 5.4.1)
Score 1200
Code Size 1553 Byte
Status AC
Exec Time 163 ms
Memory 62976 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 1200 / 1200
Status
AC × 3
AC × 26
Set Name Test Cases
Sample 00_example_01.txt, 00_example_02.txt, 00_example_03.txt
All 00_example_01.txt, 00_example_02.txt, 00_example_03.txt, 01.txt, 02.txt, 03.txt, 04.txt, 05.txt, 06.txt, 07.txt, 08.txt, 09.txt, 10.txt, 11.txt, 12.txt, 13.txt, 14.txt, 15.txt, 16.txt, 17.txt, 18.txt, 19.txt, 20.txt, 21.txt, 22.txt, 23.txt
Case Name Status Exec Time Memory
00_example_01.txt AC 72 ms 62848 KB
00_example_02.txt AC 72 ms 62848 KB
00_example_03.txt AC 140 ms 62976 KB
01.txt AC 72 ms 62848 KB
02.txt AC 93 ms 62976 KB
03.txt AC 71 ms 62848 KB
04.txt AC 73 ms 62848 KB
05.txt AC 71 ms 62848 KB
06.txt AC 72 ms 62848 KB
07.txt AC 72 ms 62848 KB
08.txt AC 72 ms 62848 KB
09.txt AC 151 ms 62976 KB
10.txt AC 72 ms 62848 KB
11.txt AC 72 ms 62848 KB
12.txt AC 149 ms 62976 KB
13.txt AC 72 ms 62848 KB
14.txt AC 123 ms 62976 KB
15.txt AC 77 ms 62976 KB
16.txt AC 123 ms 62976 KB
17.txt AC 115 ms 62976 KB
18.txt AC 148 ms 62976 KB
19.txt AC 143 ms 62976 KB
20.txt AC 110 ms 62976 KB
21.txt AC 71 ms 62848 KB
22.txt AC 163 ms 62976 KB
23.txt AC 71 ms 62848 KB