编程之美挑战赛 大神与三位小伙伴 题解

转载请注明：http://blog.csdn.net/jiangshibiao/article/details/23437025

【原题】

题目2 : 大神与三位小伙伴

时间限制: 2000ms

单点时限: 1000ms

内存限制: 256MB

描述

L国是一个有着优美景色且物产丰富的国家，很多人都喜欢来这里旅游并且喜欢带走一些纪念品，大神同学也不例外。距离开L国的时间越来越近了，大神同学正在烦恼给她可爱的小伙伴们带什么纪念品好，现在摆在大神同学面前的有三类纪念品A, B, C可以选择，每类纪念品各有N种。其中种类为A_i, B_i, C_i的纪念品价值均为i, 且分别有N+1-i个剩余。现在大神同学希望在三类纪念品中各挑选一件然后赠送给她的三名可爱的小伙伴，但是她又不希望恰好挑出来两件价值相同的纪念品，因为这样拿到相同价值纪念品的两位小伙伴就会认为大神同学偏袒另一位小伙伴而不理睬她超过一星期。现在，大神同学希望你买到的三件纪念品能让三位小伙伴都开心并且不和她闹别扭，她想知道一共有多少种不同挑选的方法？

因为方案数可能非常大，大神同学希望知道挑选纪念品的方案数模10^9+7之后的答案。

输入

第一行包括一个数T，表示数据的组数。

接下来包含T组数据，每组数据一行，包括一个整数N。

输出

对于每组数据，输出一行“Case x: ”，其中x表示每组数据的编号(从1开始)，后接一个数，表示模10^9+7后的选择纪念品的方案数。

数据范围

小数据：

1<=T<=10

1<=N<=100

大数据：

1<=T<=1000

1<=N<=10^18

样例解释

对于第二组数据，合法的方案有以下几种，(X,Y,Z)表示选择了A类纪念品中价值为X的，B类纪念品中价值为Y的，C类纪念品中价值为Z的。

(1,1,1): 3*3*3=27种

(1,2,3): 3*2*1=6种

(1,3,2): 3*1*2=6种

(2,1,3): 2*3*1=6种

(2,2,2): 2*2*2=8种

(2,3,1): 2*1*3=6种

(3,1,2): 1*3*2=6种

(3,2,1): 1*2*3=6种

(3,3,3): 1*1*1=1种

一共27+6+6+6+8+6+6+6+1=72种选择纪念品的方案

注意，如(1,1,2), (2,3,3), (3,1,3)都因为恰好选择了两件价值相同的纪念品，所以并不是一种符合要求的纪念品选择方法。

样例输入

2
1
3

样例输出

Case 1: 1
Case 2: 72

【分析】先码了一个暴力。

#include
using namespace std;
const int mo=1000000007;
int T,n,a,b,c,p;
typedef long long ll;
ll ans;
int main()
{
  scanf("%d",&T);
  for (p=1;p<=T;p++)
  {
    ans=0;
    scanf("%d",&n);
    for (a=1;a<=n;a++)
      for (b=1;b<=n;b++)
        for (c=1;c<=n;c++)
        {
          if (a==b||a==c||b==c)
          {
            if (a!=b||a!=c||b!=c) continue;
          }
          ans=(ans+((n-a+1)%mo)*((n-b+1)%mo)*((n-c+1)%mo)%mo)%mo;
        }
    printf("Case %d: %d\n",p,ans);
  }
  return 0;
}

然后开始想标程。开始以为是矩阵乘法加速。

先不考虑重复的情况，设答案是S1，S1=(1+2+3+...+n)^3=(n*(n+1)/2)^3。

再考虑有两个相同的情况（不考虑三个相同的），设答案是S2，S2=(1^2+2^2+3^2+...+n^2)*(1+2+3+..+n)=(n*(n+1)/2)*(n*(n+1)*(2*n+1)/6)。因为可能是a=b,a=c,b=c,所以最后要乘3。

再考虑三个相同的情况，设答案是S3，S3=1^3+2^3+3^3+...+n^3=(1+2+3+..+n)^2=(n*(n+1)/2)^2。

那么答案S=S1-S2*3+S3*3。因为要除以2，还要取模，所以要用逆元来求。

【应回复要求】AC截图

【代码】

#include
using namespace std;
typedef long long ll;
ll now,test,p=1000000007,N=500000004,JSB,ans,n,t;
int main()
{
  scanf("%I64d",&test);
  for (now=1;now<=test;now++)
  {
    scanf("%I64d",&n);
    JSB=(1ll*(n%p)*((n+1)%p)*N%p)%p;
    if (JSB+2ll-2*n<0)
    {
      t=(-JSB-2ll+2*n)/p+1;
    }
    ans=(1ll*JSB*JSB)%p*((JSB+2ll-2*n+t*p)%p)%p;
    printf("Case %I64d: %I64d\n",now,ans);
  }
  return 0;
}

真是抱歉，因为运算的时候括号没有注意，导致出现了一点问题。感谢下面的大神批评指正。现在代码已经改过。（就是把红括号给去掉）。和其他大神的AC程序拍过，应该不存在问题了。（当然大神们请对着括号的问题不吝赐教）