题意:现在要将5种型号的衣服分发给xn个参赛者,然后给出每个参赛者所需要的衣服的尺码的大小范围,在该尺码范围内的衣服该选手可以接受,再给出这5种型号衣服各自的数量,问是否存在一种分配方案使得每个选手都能够拿到自己尺码范围内的衣服.
可以通过建立超级原点和汇点来求。
真恶心,题目的输入形式不注意,还有初始化没注意,TLE 了几次。。。。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<set>
#include<cstdio>
#include<string>
#include<map>
using namespace std;
const int maxn=505;
//const int maxm=300006;
const int inf=1<<30;
map<char,int>mp;
struct node
{
int v,next;
int val;
}s[maxn*maxn*2];
int level[maxn];//顶点的层次
int p[maxn];
int que[maxn*10];//BFS中用于遍历的顶点,DFS求增广中记录边
int out[10*maxn];//DFS用于几乎定点的分支
int ind;
inline void insert(int x,int y,int z)
{
s[ind].v=y;
s[ind].val=z;
s[ind].next=p[x];
p[x]=ind++;
s[ind].v=x;
s[ind].val=0;
s[ind].next=p[y];
p[y]=ind++;
}
int n,f,d;
void init()
{
ind=0;
memset(p,-1,sizeof(p));
}
int max_flow(int n,int source,int sink)
{
int ret=0;
int h=0,r=0;
while(1)//DFS
{
int i;
for(i=0;i<=n;++i)
level[i]=0;
h=0,r=0;
level[source]=1;
que[0]=source;
while(h<=r)//BFS
{
int t=que[h++];
for(i=p[t];i!=-1;i=s[i].next)
{
if(s[i].val&&level[s[i].v]==0)
{
level[s[i].v]=level[t]+1;
que[++r]=s[i].v;
}
}
}
//topset=r;//记录原点的集合个数
if(level[sink]==0)break;//找不到汇点
for(i=0;i<=n;++i)
out[i]=p[i];
int q=-1;
while(1)
{
if(q<0)
{
int cur=out[source];
for(;cur!=-1;cur=s[cur].next)
{
if(s[cur].val&&out[s[cur].v]!=-1&&level[s[cur].v]==2)
{
break;
}
}
if(cur>=0)
{
que[++q]=cur;
out[source]=s[cur].next;
}
else
{
break;
}
}
int u=s[que[q]].v;
if(u==sink)//一条增广路
{
int dd=inf;
int index=-1;
for(i=0;i<=q;i++)
{
if(dd>s[que[i]].val)
{
dd=s[que[i]].val;
index=i;
}
}
ret+=dd;
//cout<<ret<<endl;
for(i=0;i<=q;i++)
{
s[que[i]].val-=dd;
s[que[i]^1].val+=dd;
}
for(i=0;i<=q;i++)
{
if(s[que[i]].val==0)
{
q=index-1;
break;
}
}
}
else
{
long cur=out[u];
for(;cur!=-1;cur=s[cur].next)
{
if(s[cur].val&&out[s[cur].v]!=-1&&level[u]+1==level[s[cur].v])
{
break;
}
}
if(cur!=-1)
{
que[++q]=cur;
out[u]=s[cur].next;
}
else
{
out[u]=-1;
q--;
}
}
}
}
return ret;
}
char ch1[20],ch2[20];
int t;
int i,j,k;
int main()
{
mp['S']=1;mp['M']=2;
mp['L']=3;mp['X']=4;
mp['T']=5;
while(scanf("%s",ch1))
{
if(!strcmp(ch1,"ENDOFINPUT"))
break;
scanf("%d",&n);
ind=0;
for( i=0;i<=2+n+7;i++)
p[i]=-1;
t=n+6;
for( i=1;i<=n;i++)
{
insert(0,i,1);
scanf("%s",ch2);
int m1=ch2[0],m2=ch2[1];
for( j=mp[m1];j<=mp[m2];j++)
insert(i,j+n,1);
}
int num=0;
for(i=1;i<=5;i++)
{
int tt;
scanf("%d",&tt);
insert(i+n,t,tt);
num+=t;//开始写的num++,晕死,不来动脑子的
}
scanf("%s",ch1);
if(num<n)//剪枝
printf("I'd rather not wear a shirt anyway...\n");
else
{
num=max_flow(2+n+5,0,t);
if(num<n)
printf("I'd rather not wear a shirt anyway...\n");
else printf("T-shirts rock!\n");
}
}
return 0;
}
poj 2584(多重匹配用--dinic模板题)
最新推荐文章于 2019-01-08 01:52:54 发布