题意:给你n个人每一个人手里有一个id,然后给你两个数a和b。让你把n个人分为两组。条件是 一组人手里的id和等于a 另一组人的id和等于b,这里的和是指加起来之后对9取余,假设sum等于0 则sum等于9 否则sum = sum。另一种情况也能够 就是全部人的id和等于a 或者等于b 相当于分为一组。
思路:首先 假设能找到满足题意的解。一定满足a和b的和等于n个人的标号的和
然后 数位dp,dp[i][j]表示在第i个数字的时候 前面i个数字能组成j而且包含arr[i]的方案数。
状态转移方程:dp[i][j] = dp[i-1][j] + dp[i-1][cnt],cnt = j - arr[i](这个应该能理解吧),cnt<=0时要加上9;
另外 假设a和b的和不等于n个人的标号的和还要考虑是否满足 a == sum或b == sum;
代码:
#include <algorithm> #include <iostream> #include <sstream> #include <cstdlib> #include <cstring> #include <iomanip> #include <cstdio> #include <string> #include <bitset> #include <vector> #include <queue> #include <stack> #include <cmath> #include <list> #include <map> #include <set> #define sss(a,b,c) scanf("%d%d%d",&a,&b,&c) #define mem1(a) memset(a,-1,sizeof(a)) #define mem(a) memset(a,0,sizeof(a)) #define ss(a,b) scanf("%d%d",&a,&b) #define s(a) scanf("%d",&a) #define p(a) printf("%d ", a) #define INF 0x3f3f3f3f #define w(a) while(a) #define PI acos(-1.0) #define LL long long #define eps 10E-9 #define N 1000000+20 #define mod 258280327 const int SIGMA_SIZE=26; const int MAXN=100010; const int MAXNODE=600010; using namespace std; void mys(int& res) { int flag=0; char ch; while(!(((ch=getchar())>='0'&&ch<='9')||ch=='-')) if(ch==EOF) res=INF; if(ch=='-') flag=1; else if(ch>='0'&&ch<='9') res=ch-'0'; while((ch=getchar())>='0'&&ch<='9') res=res*10+ch-'0'; res=flag?-res:res; } void myp(int a) { if(a>9) myp(a/10); putchar(a%10+'0'); } /*************************THE END OF TEMPLATE************************/ int arr[100009]; int dp[100009][10]; int sum_mod(int x, int y){ int ans = x + y; ans %= 9; if(!ans) return 9; return ans; } int main(){ int t, n, a, b; s(t); w(t--){ mem(dp); sss(n, a, b); int sum = 0; for(int i=1; i<=n; i++){ s(arr[i]); sum = sum_mod(sum, arr[i]); } if(sum != sum_mod(a, b)){ int ans = 0; if(a == sum) ans ++; if(b == sum) ans ++; p(ans % mod); } else{ dp[1][arr[1]] = 1; for(int i=2; i<=n; i++){ for(int j=1; j<=9; j++){ int cnt = j - arr[i] <= 0? j - arr[i] + 9: j-arr[i]; dp[i][j] = (dp[i-1][j] + dp[i-1][cnt]) % mod; } } p((dp[n][a]+dp[n][b])%mod); } } return 0; }