算法:差分约束+spfa最长路
难度:NOIP
很容易看出来这道题是差分约束,对于本题来说,因为至少一个糖果,所以add(0,i,1) 注意数组大小!!!
一次SPFA之后,直接输出dis数组总和,即为总糖果数。
注意:
1、判断自环
2、不开long long见祖宗
3、数组大小!!!
代码如下:
#include <cstdio>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <cmath>
#include <deque>
#define ll long long
#define N 100005
using namespace std;
struct node
{
int next;
int to;
int val;
}edge[N<<2];//下面还有n条边啊啊啊
int head[N],cir[N];
bool vis[N];
int cnt=1;
void init()
{
memset(head,-1,sizeof(head));
cnt=1;
}
void add(int u,int v,int w)
{
edge[cnt].next=head[u];
edge[cnt].to=v;
edge[cnt].val=w;
head[u]=cnt++;
}
int dis[N];
int n,k;
bool spfa(int rt)
{
//memset(dis,0,sizeof(dis));多清数组会导致TLE
deque<int>Q;
vis[rt]=1;
dis[rt]=0;
cir[rt]=1;
Q.push_back(rt);
while(!Q.empty())
{
int u=Q.front();
Q.pop_front();
for(int i = head[u];i!=-1;i=edge[i].next)
{
int to=edge[i].to;
if(dis[to]<dis[u]+edge[i].val)
{
dis[to]=dis[u]+edge[i].val;
if(++cir[to]>n) //如果没有环,不可能有某个点dis大于n,若大于n则必有环
{
return 0;
}
if(!vis[to])
{
vis[to]=1;
if(!Q.empty()&&dis[to]>dis[Q.front()])
{
Q.push_front(to);
}else
{
Q.push_back(to);
}
}
}
}
vis[u]=0;
}
return 1;
}
void read(int &x)
{
int f=1;x=0;char s=getchar();
while(s<'0'||s>'9') s=getchar();
while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();}
x*=f;
}
int main()
{
scanf("%d%d",&n,&k);
//read(n),read(k);
init();
for(int i = 1;i <= k;i++)
{
int x,a,b;
//scanf("%d%d%d",&x,&a,&b);
read(x),read(a),read(b);
if((x%2==0)&&(a==b))//注意判断自环
{
puts("-1");
return 0 ;
}
if(x==1)
{
add(a,b,0);
add(b,a,0);
}else if(x==2)
{
add(a,b,1);
}else if(x==3)
{
add(b,a,0);
}else if(x==4)
{
add(b,a,1);
}else if(x==5)
{
add(a,b,0);
}
}
for(int i = n;i >= 1;i--) add(0,i,1);//为什么倒着加边就能A???
ll ans=0;
if(!spfa(0))
{
puts("-1");
return 0 ;
}else
{
for(int i = 1;i <= n;i++)ans += dis[i]; //dis[i]是第i个小朋友最少需要的糖果数量
printf("%lld\n",ans);
}
return 0 ;
}