[HAOI 2012] 容易题

[题目链接]

https://www.lydsy.com/JudgeOnline/problem.php?id=2751

[算法]

考虑k = 0的情况 ， 根据乘法原理 :

Ans = (n * (n + 1) / 2) ^ m

那么 ， 对于k > 0 , 只需将用一棵平衡树维护每个位置应减小的值即可

详见代码

时间复杂度 : O(NlogN)

[代码]

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
const int P = 1e9 + ;

int n , m , k;
map<int , int> mp;
map< pair<int , int> , bool> existed;

template <typename T> inline void chkmax(T &x,T y) { x = max(x,y); }
template <typename T> inline void chkmin(T &x,T y) { x = min(x,y); }
template <typename T> inline void read(T &x)
{
    T f = ; x = ;
    char c = getchar();
    for (; !isdigit(c); c = getchar()) if (c == '-') f = -f;
    for (; isdigit(c); c = getchar()) x = (x << ) + (x << ) + c - '';
    x *= f;
}
inline int exp_mod(int a , int n)
{
        int b = a , res = ;
        while (n > )
        {
                if (n & ) res = 1ll * res * b % P;
                b = 1ll * b * b % P;
                n >>= ;
        }
        return res;
}

int main()
{

        read(n); read(m); read(k);
        for (int i = ; i <= k; ++i)
        {
                int x , y;
                read(x); read(y);
                if (existed[make_pair(x , y)]) continue;
                existed[make_pair(x , y)] = true;
                mp[x] += y;
        }
        int cnt = (1ll * n * (n + ) >> ) % P ,
            rest = m - (int)mp.size();
        int ans = exp_mod(cnt , rest);
        for (map<int , int> :: iterator it = mp.begin(); it != mp.end(); it++)
                ans = (1ll * ans * ((cnt - it -> second % P) % P + P) % P) % P;
        printf("%d\n" , ans);

        return ;

}