第二次排位赛-G(dfs)

传送门：https://www.luogu.org/problemnew/show/P5118
题目大意：第一第二个奶棚都有1000的奶，周二FJ把在第一个奶棚的任意一桶奶搬到第二个奶棚，周三从第二个奶棚搬任意一桶到第一个奶棚，周四同周二，周五同周三，问周五搬完之后第一个奶棚的奶量有多少种可能。
思路：dfs，搬了的桶标记一下，搬过去的桶扔到数组里，记录桶的数量，此时桶的数量+1
代码：

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const ll M=2e3+10;
ll a[2][20],sz[2],cnt[M],tk=1000;
bool taken[2][20];
void dfs(ll floor){
	if(floor==4){
		cnt[tk]=1;
		return;
	}
	if((floor&1)==0){
		for(ll i=0;i<sz[0];i++){
			if(taken[0][i]) continue;
			tk-=a[0][i];
			taken[0][i]=1;
			a[1][sz[1]++]=a[0][i];
			dfs(floor+1);
			tk+=a[0][i];
			taken[0][i]=0;
			sz[1]--;
		}
	}
	else{
		for(ll i=0;i<sz[1];i++){
			if(taken[1][i]) continue;
			tk+=a[1][i];
			taken[1][i]=1;
			a[0][sz[0]++]=a[1][i];
			dfs(floor+1);
			tk-=a[1][i];
			taken[1][i]=0;
			sz[0]--;
		}
	}
}
ll count(){
	dfs(0);
	ll ans=0;
	for(ll i=0;i<=2000;i++)
		ans+=cnt[i];
	return ans;
}
int main(){
	sz[0]=sz[1]=10;
	for(ll i=0;i<2;i++)
		for(ll j=0;j<10;j++)
			cin>>a[i][j];
	dfs(0);
	cout<<count();
	return 0;
}