HAUTOJ 1265 Hmz 的女装

Hmz 的女装

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit:131072/131072 K (Java/Others)
Total Submission(s): 266    Accepted Submission(s): 119

Problem Description

Hmz为了女装，想给自己做一个长度为n的花环。现在有k种花可以选取，且花环上相邻花的种类不能相同。
Hmz想知道，如果他要求第l朵花和第r朵花颜色相同，做花环的方案数是多少。这个答案可能会很大，你只要输出答案对10^9+7取模的结果即可。

 

 

Input

第一行三个整数n,m,k(1≤n≤100000,1≤m≤100000,1≤k≤100000)
接下来m行，每行两个整数l,r，表示要求第l朵花和第r朵花颜色相同。保证l≠r且|(r-l) mod n| ≠1.

 

 

Output

输出m行。对于每一个询问输出一个整数，表示做花环的方案数对10^9+7取模的结果。

 

 

Sample Input

8 3 2

1 4

2 6

1 3

8 3 3

1 4

2 6

1 3

 

 

Sample Output

0

2

2

60

108

132

思路：由于这是一个花环，所以我们可以得到如果要求某两个位置花颜色相同的方案数，应该只跟这两个位置之间的距离有关。

我们可以把花环在L,R两点切开，得到了均不包含L,R两点的两条线段，设长度分别为len1,len2.由于要求L,R的花色相同，设这个花色为color,那么我们应该要让这两条线段的两个端点颜色都不能为color,那么问题就转化为求一条线段两个端点花色都不为color的方案数，由前面的分析我们可以设dp[i]为长度为i且两端花色都不为color的方案数，那么结果结果可以表示为

ans=dp[len1]*dp[len2]*k

乘以k的原因是color的选取有k种可能。

那么怎么来求dp[i]的值呢？

我们可以这样来考虑：

用dp[i][0]表示长度为i且第i朵花的颜色与第1朵花的颜色不同的方案数。

用dp[i][1]表示长度为i且第i朵花的颜色与第1朵花的颜色相同的方案数。

写出状态转移方程

dp[i][0]=dp[i-1][0](k-2)+dp[i-1][1](k-1)

dp[i][1]=dp[i-1][0]

注意：这里中间转移时不需要考虑花色不能为color(除了第一朵花)，最后再考虑由于dp[i][0]要减去最后第i朵花花色不能为color,还要再加上第i朵花颜色可以与第一朵花相同，刚好抵消，dp[i][0]即为dp[i]

两式合并可以得到

dp[i][0]=dp[i-2][0](k-1)+dp[i-1][0](k-2)

也就得到了dp[i]的递推方程

(dalao有一种递推方法：dp[i]=(k-1)^i-dp[i-1],不过我到现在还没证明出来TUT)

#include <cstdio>
#include <cmath>
#include <queue>
#include <map>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define mst(a,b) memset((a),(b),sizeof(a))
#define f(i,a,b) for(int i=(a);i<=(b);++i)
#define rush() int T;scanf("%d",&T);while(T--)

typedef long long ll;
const int maxn= 100005;
const ll mod = 1e9+7;
const double INF = 0x3f3f3f3f;
const double eps = 1e-6;

ll dp[maxn];
ll dp2[maxn];

int main()
{
    ll n,m,k;
    while(~scanf("%I64d%I64d%I64d",&n,&m,&k))
    {
        dp[0]=0;
        dp[1]=k-1;
        /*ll temp=1;
        for(int i=1;i<=n;i++)
        {
            temp=(temp*(k-1))%mod;
            dp[i]=(temp-dp[i-1]+mod)%mod;
        }*/
        for(int i=2;i<=n;i++)
        {
            dp[i]=(dp[i-2]*(k-1)%mod+dp[i-1]*(k-2)%mod)%mod;
        }
        ll l,r;
        for(int i=0;i<m;i++)
        {
            scanf("%I64d%I64d",&l,&r);
            if(l>r)
                swap(l,r);
            ll ans=dp[r-l-1]*k%mod*dp[n-(r-l)-1]%mod;
            printf("%I64d\n",ans);
        }
    }
    return 0;
}