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