haut 1265: Hmz 的女装(DP)@

1265: Hmz 的女装

时间限制: 2 秒  内存限制: 128 MB
提交: 85  解决: 18
提交  状态 

题目描述

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

输入

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

输出

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

样例输入

8 3 2
1 4
2 6
1 3
8 3 3
1 4
2 6
1 3

样例输出

0
2
2
60
108
132

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e6+7;
const int inf = 0x3f3f3f3f;
const LL mod = 1e9+7;
LL dp[N][3];
 
int main()
{
    int n, m;
    LL k;
    while(scanf("%d %d %lld", &n, &m, &k)!=EOF)
    {
        dp[0][1]=1, dp[0][0]=0;
        for(int i=1; i<=n; i++)
        {
            dp[i][0]=(((dp[i-1][1]%mod)*(k-1)%mod)%mod+((dp[i-1][0]%mod)*(k-2)%mod)%mod)%mod;
            dp[i][1]=(dp[i-1][0]);
        }
        int l, r;
        while(m--)
        {
            scanf("%d %d", &l, &r);
            if(l>r) swap(l,r);
            int x=max(r-l-1,0);
            int y=max(n-x-2,0);
            LL ans=(dp[x][0]%mod)*(dp[y][0]%mod)%mod;
            cout<<(ans%mod)*(k%mod)%mod<<endl;
        }
    }
    return 0;
}