社区讨论

已过,求解代码时间复杂度(玄关)

P4099[HEOI2013] SAO参与者 2已保存回复 1

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
1 条
当前快照
1 份
快照标识符
@mhphvnvj
此快照首次捕获于
2025/11/08 07:36
3 个月前
此快照最后确认于
2025/11/09 02:27
3 个月前
查看原帖
我觉得是 O(n3)O(n^3),但是过了。
CPP
#include<bits/stdc++.h>
using namespace std;
typedef long long ll; 
const int N=1e3+100,mod=1e9+7;
ll t,n,ans;
ll f[N][N],rf[N][N],sz[N];
ll C[N][N];
vector<int> a[N],b[N];
void add(ll &x,ll y){
	x=(x+y)%mod;
}
void fact(){
	C[0][0]=1;
	for(int i=1;i<=N-10;i++){
		C[i][0]=1;
		for(int j=1;j<=i;j++)
			C[i][j]=(C[i-1][j-1]+C[i-1][j])%mod;
	}
}
void dfs(int u,int fa){
	f[u][1]=1;
	sz[u]=1;
	for(int v:a[u]){
		if(v==fa) continue;
		dfs(v,u);
		int orz=sz[u];
		sz[u]+=sz[v];
		for(int i=1;i<=orz;i++)
			rf[u][i]=f[u][i];
		for(int i=1;i<=sz[v];i++)
			rf[v][i]=(f[v][i]+rf[v][i-1])%mod;
		for(int i=1;i<=sz[u];i++){
			f[u][i]=0; 
			for(int j=max(1ll,i+orz-sz[u]);j<=min(orz,i);j++){
				add(f[u][i],rf[u][j]*rf[v][i-j]%mod*C[i-1][j-1]%mod*C[sz[u]-i][orz-j]%mod);
			}
		}
	}
	for(int v:b[u]){
		if(v==fa) continue;
		dfs(v,u);
		int orz=sz[u];
		sz[u]+=sz[v];
		for(int i=1;i<=orz;i++)
			rf[u][i]=f[u][i];
		rf[v][sz[v]+1]=0;
		for(int i=sz[v];i>=1;i--)
			rf[v][i]=(f[v][i]+rf[v][i+1])%mod;
		for(int i=1;i<=sz[u];i++){
			f[u][i]=0;
			for(int j=max(1ll,i+orz-sz[u]);j<=min(orz,i);j++){
				add(f[u][i],rf[u][j]*rf[v][i-j+1]%mod*C[i-1][j-1]%mod*C[sz[u]-i][orz-j]%mod);
			}
		}
	}
}
int main(){
	ios::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);
	fact();
	cin>>t;
	while(t--){
		cin>>n;
		for(int i=1;i<=n;i++)
			a[i].clear(),b[i].clear();
		for(int i=1;i<n;i++){
			int x,y; char c;
			cin>>x>>c>>y;
			x++,y++;
			if(c=='>') swap(x,y);
			a[y].push_back(x);
			b[x].push_back(y);
		}
		dfs(1,0);
		ans=0;
		for(int i=1;i<=n;i++)
			add(ans,f[1][i]);
		cout<<ans<<"\n";
	}
	return 0;
}

回复

1 条回复,欢迎继续交流。

正在加载回复...