CF 补题 E. Boboniu Walks on Graph

E. Boboniu Walks on Graph

题目链接

题目意思

给一个有向图，在图上走的时候，出度为 i 的边，只能走到边权大小为第 ci 小的边上面，问c数组有多少种，要每个点都满足：从这个点开始走都可以回到这个点。
每个点的入度都小于等于9

题解

都可以回到这个点，，但是每个点的出边只能有一条，所以n个点，入度都为1。
因为入度都是小于10的所以可以暴力枚举一下c数组，然后check。
怎么check？ 只要按照枚举出来的 c 数组走，每个点的入度等于1就可以。
所以可以预处理一下，预处理入读为 i 的所有点走权值大小为 j 的时候，可以走到哪些点。
刚开始用的bitset，果断tle。改成哈希就好了。

#include
#include
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
typedef long long ll;
typedef pair<int,ll> pii;
typedef pair<ll,ll> pll;
typedef pair<double,double> pdd;
typedef unsigned long long ull;
typedef unordered_set<int>::iterator sit;
#define st first
#define sd second
#define mkp make_pair
#define pb push_back
void tempwj(){
     freopen("hash.in","r",stdin);freopen("hash.out","w",stdout);}
ll gcd(ll a,ll b){
     return b == 0 ? a : gcd(b,a % b);}
ll qpow(ll a,ll b,ll mod){
     a %= mod;ll ans = 1;while(b > 0){
     if(b & 1)ans = ans * a % mod;a = a * a % mod;b >>= 1;}return ans;}
struct cmp{
     bool operator()(const pii & a, const pii & b){
     return a.second > b.second;}};
int lb(int x){
     return  x & -x;}
//friend bool operator < (Node a,Node b)   重载
const int inf = INT_MAX;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll mod = 1e9+7;
const int maxn = 2e5+10;
const int M = 1e6 + 2;
int temp[maxn];
int n,k;
// bitset flag;
// bitset ss;
// bitset val[10][10];
int key[maxn];
int val[10][10];
int ans =0 ;
int flag;
void dfs(int pos)
{
     
	if(pos > k)
	{
     
		// ss.reset();
		int ss = 0;
		for (int i = 1; i <= k; i ++ )
		{
     
			ss += val[i][temp[i]];
		}
		// if(ss == flag)
		// {
     
		// 	// for (int i = 1; i <= k; i ++ )
		// 	// {
     
		// 	// 	printf("%d ",temp[i]);
		// 	// }
		// 	// printf("  1111111\n");
		// 	ans ++ ;
		// }
		if(ss == flag)
			ans ++ ;
		return;
	}
	for (int i = 1; i <= pos; i ++ )
	{
     
		temp[pos] = i;
		dfs(pos + 1);
	}
}
std::vector<pii> vv[maxn];
int du[maxn];
int main()
{
     
	int m;
	scanf("%d%d%d",&n,&m,&k);
	for (int i =1 ; i<= n; i ++ )
	{
     
		key[i] = rand();
		flag += key[i];
	}
	for (int i = 1; i <= m; i ++ )
	{
     
		int x,y,v;
		scanf("%d%d%d",&x,&y,&v);
		du[x] ++ ;
		vv[x].pb(mkp(v,y));
	}
	for (int i =1 ; i <= n; i ++ )
		sort(vv[i].begin(),vv[i].end());
	for (int i = 1; i <= n; i ++ )
	{
     
		for (int j = 0; j < vv[i].size(); j ++ )
		{
     
			val[du[i]][j + 1] += key[vv[i][j].sd];
		}
	}
	dfs(1);
	printf("%d\n",ans);
}