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

P1640 [SCOI2010]连续攻击游戏

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

#include 

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 
#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;
}

你可能感兴趣的:(图论,图论,深度优先,算法)