题目:http://acm.timus.ru/problem.aspx?space=1&num=1903
题目大意:组合数,给你n个数,从中要选t个数,排成非递减序列,第k个数一定要选,且它在选出来的序列中处在第x个位置,问你有多少种选法?
思路:思路应该是比较简单的,就是比他小的、大的,相等的统计出来,然后就是组合数了,枚举一下,小的里面取i个,大的取j个,那么相等的就是t-i-j了,然后三个C相乘,再全加起来就好了。空间开5000*2500,应该差不多正好。
我们先开始时直接算了出现的三种C,从前推后的,现在才知道推不了,因为要*一个数再 /k,但是mod之后就除不了了,而 + 、- 没事。。 循环的时候要特别注意三者的限制条件,这里错了很多次,特别是 i+k+1<pos 这一句,枚举i、j的时候也是,都要取小的,反正是WA了很多遍,。。 唉,不说了,说多了都是泪啊。。 = =
代码如下:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN = 5010 ;
const int MOD = 1e9+7;
typedef __int64 lld;
int a[MAXN];
int c[MAXN][MAXN>>1];
void init()
{
c[0][0] = 1;
c[1][0] = 1;
c[1][1] = 1;
for(int i =2;i<=5000;i++)
{
c[i][0] = 1;
for(int j = 1;j<=i/2;j++)
{
c[i][j] = (c[i-1][min(j-1,i-j)]+c[i-1][min(j,i-1-j)])%MOD;
}
}
}
int main()
{
init();
int n,remain;
while(~scanf("%d%d",&n,&remain))
{
for(int i = 1;i<=n;i++)
scanf("%d",&a[i]);
int iden,pos;
scanf("%d%d",&iden,&pos);
int small =0,big=0,equ =0 ;
for(int i = 1;i<=n;i++)
if(a[i]<a[iden])
small ++;
else if(a[i]>a[iden])
big++;
else equ ++;
equ--;
//printf("%d,%d,%d\n",small,big,equ);
int left = pos-1;
int right = remain-pos ;
lld ans = 0;
for(int i = 0;i<=min(left,small);i++)
for(int j = 0;j<=min(right,big);j++)
{
int k = remain - i - j -1 ;
if(i+k+1<pos) break;
if(k>equ) continue;
if(k<0) break;
lld tmp = 1;
tmp = tmp*c[small][min(i,small-i)]*c[big][min(j,big-j)]%MOD;
tmp = tmp*c[equ][min(k,equ-k)]%MOD ;
ans = (tmp +ans)%MOD;
//printf("i = %d, j = %d,k =%d,ans = %I64d\n",i,j,k,ans);
}
printf("%I64d\n",ans);
}
return 0;
}