BZOJ 2330 查分約束

點擊打開鏈接

題意：中文

思路：根據條件構造模型即可

#include <queue>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=100010;
int dis[maxn],cnt[maxn],head[maxn],n,m,k;
bool vis[maxn];
struct edge{
    int to,w,next;
}E[maxn*10];
void add_edge(int u,int v,int w){
    E[k].to=v;E[k].w=w;E[k].next=head[u];head[u]=k++;
}
bool spfa(){
    queue<int>que;
    for(int i=1;i<=n;i++){
        dis[i]=1;vis[i]=1;cnt[i]=1;
        que.push(i);
    }
    while(!que.empty()){
        int t=que.front();que.pop();
        vis[t]=0;
        for(int i=head[t];i!=-1;i=E[i].next){
            if(dis[t]+E[i].w>dis[E[i].to]){
                dis[E[i].to]=dis[t]+E[i].w;
                if(++cnt[E[i].to]>=n) return 0;
                if(!vis[E[i].to]){
                    vis[E[i].to]=1;
                    que.push(E[i].to);
                }
            }
        }
    }
    return 1;
}
int main(){
    int a,b,op;
    while(scanf("%d%d",&n,&m)!=-1){
        k=0;memset(head,-1,sizeof(head));
        int flag=0;
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&op,&a,&b);
            if(op==2||op==4){
                if(a==b) flag=1;
            }
            if(op==1) add_edge(a,b,0),add_edge(b,a,0);
            if(op==2) add_edge(a,b,1);
            if(op==3) add_edge(b,a,0);
            if(op==4) add_edge(b,a,1);
            if(op==5) add_edge(a,b,0);
        }
        if(flag){
            printf("-1\n");
            continue;
        }
        int ans=spfa();
        if(ans==1){
            ll sum=0;
            for(int i=1;i<=n;i++) sum+=(ll)dis[i];
            printf("%lld\n",sum);
        }else printf("-1\n");
    }
    return 0;
}