关于set和map
查看原帖
关于set和map
235901
Always_Remember_It楼主2024/7/25 10:52
using namespace std;
#define int long long
const int N=1e6+10;
int n,a[N],b[N],x[N],y[N],cnt,head[N],tot;
int dp[N][2];
bool vis[N];
struct Edge{
	int next,to;
}e[N];
set<int> s[N];
//map<int,map<int,int> > mp;
void adde(int u,int v){
	e[++tot].next=head[u];
	e[tot].to=v;
	head[u]=tot;
	return;
}
void dfs1(int u,int s){
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to;
		if(v==s){
			++cnt;
			x[cnt]=u,y[cnt]=v;
			return;
		}
		if(vis[v]) continue;
		vis[v]=1;
		dfs1(v,s);
	}
	return;
}
void dfs2(int u,int y){
	if(u!=y) dp[u][1]+=a[u];
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to;
		dfs2(v,y);
		dp[u][0]+=max(dp[v][0],dp[v][1]);
		if(u!=y) dp[u][1]+=dp[v][0];
	}
	return;
}
inline int read(){
	int s=0;
	char ch=getchar();
	while(ch<'0'||ch>'9'){
		ch=getchar();
	}
	while(ch>='0'&&ch<='9'){
		s=(s<<1)+(s<<3)+(ch^48);
		ch=getchar();
	}
	return s;
}
signed main(){
	n=read();
	for(int i=1;i<=n;i++){
		a[i]=read(),b[i]=read();
		adde(b[i],i);
	}
	for(int i=1;i<=n;i++){
		if(!vis[i]) dfs1(i,i);
	}
	for(int i=1;i<=cnt;i++){
		s[x[i]].insert(y[i]);
     //mp[x[i]][y[i]]=1;
	}
	tot=0;
	memset(head,0,sizeof(head));
	memset(e,0,sizeof(e));
	for(int i=1;i<=n;i++){
		if(!s[b[i]].count(i)) adde(b[i],i);
      //if(!mp[b[i]][i]) adde(b[i],i);
	}
	int ans=0;
	for(int i=1;i<=cnt;i++){
		dfs2(y[i],y[i]);
		int maxn=dp[y[i]][0];
		memset(dp,0,sizeof(dp));
		dfs2(y[i],x[i]);
		maxn=max(maxn,max(dp[y[i]][1],dp[y[i]][0]));
		ans+=maxn;
	}
	printf("%lld\n",ans);
	return 0; 
}
为什么用set能过,但是用map会MLE,请问一下set和map的一维,二维空间怎么算(set<int>s[N],map<int,int>mp[N]是填多少算多少还是开了N就算N
2024/7/25 10:52
加载中...