Submission #403175

#	Time	Username	Problem	Language	Result	Execution time	Memory
403175		MohamedAhmed04	Anagramistica (COCI21_anagramistica)	C++14	110 / 110	16 ms	16332 KiB

anagramistica

#include <bits/stdc++.h>

using namespace std ;

const int mod = 1e9 + 7 ;

int Add(int x , int y)
{
	int z = x + y ;
	if(z >= mod)
		z -= mod ;
	return z ;
}

int Sub(int x , int y)
{
	int z = x - y ;
	if(z < 0)
		z += mod ;
	return z ;
}

int Mul(int x , int y)
{
	return (x * 1ll * y) % mod ;
}

int powlog(int base , int power)
{
	if(power == 0)
		return 1 ;
	int x = powlog(base , power / 2) ;
	x = Mul(x , x) ;
	if(power & 1)
		x = Mul(x , base) ;
	return x ;
}

int modinv(int x)
{
	return powlog(x , mod-2) ;
}

int Div(int x , int y)
{
	return Mul(x , modinv(y)) ;
}

struct combination
{
    vector<int>fact , inv ;
    combination(int sz) : fact(sz + 1) , inv(sz + 1)
    {
        fact[0] = 1 ;
        inv[0] = 1 ;
        for(int i = 1 ; i <= sz ; ++i)
            fact[i] = Mul(fact[i-1] , i) ;
        inv[sz] = modinv(fact[sz]) ;
		for(int i = sz-1 ; i >= 1 ; --i)
		    inv[i] = Mul(inv[i+1] , i+1) ;
    }
    int choose(int n , int k) const
    {
        if(k < 0 || n < k)
            return 0 ;
        return Mul(Mul(fact[n] , inv[k]) , inv[n - k]) ;
    }
};

const int MAX = 2000 + 10 ;

combination comb(MAX) ;

string arr[MAX] ;
int n , k ;

map<string , int>mp ;
vector<int>v ;

int dp[MAX][MAX] ;

int solve(int idx , int rem)
{
	if(idx == n)
		return (rem == 0) ;
	int &ret = dp[idx][rem] ;
	if(ret != -1)
		return ret ;
	ret = solve(idx+1 , rem) ;
	int sum = 0 ; 
	for(int i = 0 ; i < v[idx] ; ++i)
	{
		sum += i ;
		if(sum > rem)
			break ;
		ret = Add(ret , Mul(solve(idx+1 , rem - sum) , comb.choose(v[idx] , i+1))) ;
	}
	return ret ;
}

int main()
{
	memset(dp , -1 , sizeof(dp)) ;
	ios_base::sync_with_stdio(0) ;
	cin.tie(0) ;
	cin>>n>>k ;
	for(int i = 0 ; i < n ; ++i)
		cin>>arr[i] ;
	for(int i = 0 ; i < n ; ++i)
	{
		sort(arr[i].begin() , arr[i].end()) ;
		mp[arr[i]]++ ;
	}
	for(auto &i : mp)
		v.push_back(i.second) ;
	n = v.size() ;
	return cout<<solve(0 , k)<<"\n" , 0 ;
}		

• Subtask 1: 10 / 10
• Subtask 2: 30 / 30
• Subtask 3: 70 / 70