这种多对一的选择问题一般都是二分图,这道题也显然,只需要把三种运算的结果离散化一下跑一个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