BZOJ 4429 小学数学 - 最大匹配+离散化

这种多对一的选择问题一般都是二分图，这道题也显然，只需要把三种运算的结果离散化一下跑一个hungary就好了，奇怪的是这么大的数据还可以好的这么快。。。

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<algorithm>

using namespace std;

const int maxn=25005;

struct edge
{
    int to,next;
}e[maxn];

pair<int,int>q[maxn];
int n,cnt,T;
long long disc[maxn];
int head[maxn],vst[maxn],match[maxn];

void insert(int a,int b)
{
    static int ecnt=0;
    e[++ecnt].to=b;e[ecnt].next=head[a];head[a]=ecnt;
}
bool find(int x)
{
    for(int i=head[x];i;i=e[i].next)
    {
        int y=e[i].to;
        if(vst[y]==T)continue;
        vst[y]=T;
        if(!match[y]||find(match[y]))
        {
            match[y]=x;
            return true;
        }
    }
    return false;
}
void print(int i,long long val)
{
    printf("%d ",q[i].first);
    if(q[i].first+q[i].second==val)
        putchar('+');
    else if(q[i].first-q[i].second==val)
        putchar('-');
    else putchar('*');
    printf(" %d = %lld\n",q[i].second,val);
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        long long a,b;
        scanf("%lld%lld",&a,&b);
        q[i].first=a;q[i].second=b;
        disc[++cnt]=a+b;
        disc[++cnt]=a-b;
        disc[++cnt]=1LL*a*b;
    }
    sort(disc+1,disc+cnt+1);
    cnt=unique(disc+1,disc+cnt+1)-(disc+1);
    for(int i=1;i<=n;i++)
    {
        insert(lower_bound(disc+1,disc+cnt,q[i].first+q[i].second)-disc,i);
        insert(lower_bound(disc+1,disc+cnt,q[i].first-q[i].second)-disc,i);
        insert(lower_bound(disc+1,disc+cnt,1LL*q[i].first*q[i].second)-disc,i);
    }
    for(int i=1;i<=cnt;i++)
        T=i,find(i);
    for(int i=1;i<=n;i++)if(!match[i])
    {
        puts("impossible");
        return 0;
    }
    for(int i=1;i<=n;i++)
        print(i,disc[match[i]]);
    return 0;
}

HOME Back