二分图的时间戳优化+并查集变型

Posted 2022-03-19 钟钟终

P1640 [SCOI2010]连续攻击游戏

用used数组记录now(now++),不需要每一轮都是用memset判别。

#include <bits/stdc++.h>

using namespace std;
const int maxn=4e6+5;
int n,link[maxn],ans,cnt,now,head[maxn];
int used[maxn];
struct node

    int to,nxt;
e[maxn];
void add(int u,int v)

    e[++cnt].to=v;
    e[cnt].nxt=head[u];
    head[u]=cnt;

int dfs(int u)

    for(int i=head[u];i;i=e[i].nxt)
    
        int v=e[i].to;
        if(used[v]!=now)
        
            used[v]=now;
            if(link[v]==-1||dfs(link[v]))
            
                link[v]=u;
                return 1;
            
        
    
    return 0;

void hungary()

    memset(link,-1,sizeof(link));
    for(int i=1;i<=10000;i++)
    
        now++;
        if(dfs(i))
            ans++;
        else
            break;
    

int main()

    int n;cin>>n;
    for(int i=1;i<=n;i++)
    
        int u,v;cin>>u>>v;
        add(u,i);add(v,i);
    
    hungary();
    cout<<ans<<endl;
    return 0;

P1525 [NOIP2010 提高组] 关押罪犯

将罪恶值从大到小排序，两个监狱分别存放有罪恶联系的两个人；将自己的敌人放到对面，将敌人的敌人放到自己这来拆分。若拆分后，自己所在监狱出现不可避免地罪恶值，则是答案。
一种特殊情况：若所有犯人都被拆分到两个监狱，没有冲突，则输出0。
技巧：再多跑一轮，e[i].x，e[i].y和e[i].z都为0，符合特殊情况。

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
const int maxn=2e5+5;
int a[maxn],b[maxn],f[maxn],n,m;
struct node

    int x,y,z;
e[maxn];
int r_find(int r)

    if(r!=f[r])
        f[r]=r_find(f[r]);
    return f[r];

void update(int x,int y)

    f[r_find(x)]=r_find(y);

bool cmp(node a1,node a2)

    return a1.z>a2.z;

signed main()

    scanf("%lld%lld",&n,&m);
    for(int i=1;i<=n;i++) f[i]=i;
    for(int i=1;i<=m;i++)
    
        scanf("%lld%lld%lld",&e[i].x,&e[i].y,&e[i].z);
    
    sort(e+1,e+m+1,cmp);
    for(int i=1;i<=m+1;i++)
    
        if(r_find(e[i].x)==r_find(e[i].y))
        
            cout<<e[i].z<<endl;break;
        
        if(!b[e[i].x]) b[e[i].x]=e[i].y;
        else
            update(b[e[i].x],e[i].y);
        if(!b[e[i].y]) b[e[i].y]=e[i].x;
        else
            update(e[i].x,b[e[i].y]);
    
    return 0;