首页 > 其他 > 详细

P1993 小K的农场(差分约束)

时间:2019-06-09 12:36:14      阅读:99      评论:0      收藏:0      [点我收藏+]

小K的农场


题目描述

小K在MC里面建立很多很多的农场,总共n个,以至于他自己都忘记了每个农场中种植作物的具体数量了,他只记得一些含糊的信息(共m个),以下列三种形式描述:
    农场a比农场b至少多种植了c个单位的作物,
    农场a比农场b至多多种植了c个单位的作物,
    农场a与农场b种植的作物数一样多。
但是,由于小K的记忆有些偏差,所以他想要知道存不存在一种情况,使得农场的种植作物数量与他记忆中的所有信息吻合。
输入输出格式
输入格式:

第一行包括两个整数 n 和 m,分别表示农场数目和小 K 记忆中的信息数目。

接下来 m 行:

如果每行的第一个数是 1,接下来有 3 个整数 a,b,c,表示农场 a 比农场 b 至少多种植了 c 个单位的作物。

如果每行的第一个数是 2,接下来有 3 个整数 a,b,c,表示农场 a 比农场 b 至多多种植了 c 个单位的作物。如果每行的第一个数是 3,接下来有 2 个整数 a,b,表示农场 a 种植的的数量和 b 一样多。

输出格式:

如果存在某种情况与小 K 的记忆吻合,输出“Yes”,否则输出“No”。

输入输出样例
输入样例#1: 

3 3
3 1 2
1 1 3 1
2 2 3 2

输出样例#1: 
Yes

说明

对于 100% 的数据保证:1 ≤ n,m,a,b,c ≤ 10000。

差分约束可以求解最短路和最长路。这道题也是差分约束的模板题。
根据:

 农场a比农场b至少多种植了c个单位的作物,
    农场a比农场b至多多种植了c个单位的作物,
    农场a与农场b种植的作物数一样多。

建立条件。
由题可知,农场a比农场b至少多种植了c个单位的作物,所以Xa-Xb>=cXb-Xa<=-c, 农场a比农场b至多多种植了c个单位的作物,所以Xa-Xb<=c,农场a与农场b种植的作物数一样多,所以Xa==Xb,则Xa-Xb<=cXb-Xa<=c
然后用SPFA。(不要连错了权值)

#include<bits/stdc++.h>
using namespace std;

const int maxn=11000;
const int inf=0x3f3f3f3f;

int n,m;

struct node{
    int v,w;
    node(){ }
    node(int _v,int _w){
        v=_v;
        w=_w;
    }
};

vector <node> g[maxn];
int dst[maxn];
queue <int> qu;
bool inq[maxn];
int cnt[maxn];


int add(int u,int v,int w){
    g[u].push_back(node(v,w));
}

bool spfa(int u){
    memset(dst,inf,sizeof dst);
//  memset(cnt,0,sizeof cnt);
    dst[u]=0;
    qu.push(u);
    inq[u]=1;
    while(!qu.empty()){
        u=qu.front();
        qu.pop();
        inq[u]=0;
        for(int i=0;i<g[u].size();i++){
            int v=g[u][i].v;
            int w=g[u][i].w;
            if(dst[v]>dst[u]+w){
                dst[v]=dst[u]+w;
                if(!inq[v]){
                    qu.push(v);
                    inq[v]=1;
                    cnt[v]++;
                    if(cnt[v]>n){
                        return 0;   
                    }
                }
            }
        }
    }
    return 1;
}


int main(){
    cin >> n >> m;
     for(int i=1;i<=n;i++){
        add(0,i,0);
    }
    for(int i=0;i<m;i++){
        int d,a,b,c;
        cin >> d;
        if(d==1){
            cin >>a>>b>>c;
            g[a].push_back(node(b,-c));
        }else if(d==2){
            cin >>a>>b>>c;
            g[b].push_back(node(a,c));
        }else{
            cin >>a>>b;
            g[a].push_back(node(b,0));
            g[b].push_back(node(a,0));
        }
    }
    
   
    if(spfa(0)){
        cout << "Yes";
    }else{
        cout << "No";
    }
    return 0;
}

但实际上,它只能得到60分。四组TLE。。。
于是,就涉及到另外一个数据结构,双向队列。双向队列有队列和栈的性质。可以从两端入队,弹出。在这道题里,我们用if判断,在队列后端放入较大的值,前端放入较小的值分别用back和front访问最后一个元素和第一个元素。

#include<bits/stdc++.h>
using namespace std;

const int maxn=11000;
const int inf=0x3f3f3f3f;

int n,m;

struct node{
    int v,w;
    node(){ }
    node(int _v,int _w){
        v=_v;
        w=_w;
    }
};

vector <node> g[maxn];
int dst[maxn];
deque<int> qu;//双向队列 
bool inq[maxn];
int cnt[maxn];


int add(int u,int v,int w){
    g[u].push_back(node(v,w));
}

bool spfa(int u){
    memset(dst,inf,sizeof dst);//初始化 
//  memset(cnt,0,sizeof cnt);
    dst[u]=0;
    qu.push_back(u);//双向队列的访问最后一个元素写法 
    inq[u]=1;
    while(!qu.empty()){
        u=qu.front();//双向队列的访问第一个元素写法 
        qu.pop_front();
        if(dst[qu.front()]>dst[qu.back()]){
            swap(qu.front(),qu.back());//交换
        }
        inq[u]=0;
        for(int i=0;i<g[u].size();i++){
            int v=g[u][i].v;
            int w=g[u][i].w;
            if(dst[v]>dst[u]+w){
                dst[v]=dst[u]+w;
                if(!inq[v]){
                    if(dst[v]<dst[qu.front()]){//判断,比较大小 
                        qu.push_front(v);//插入队首 
                    }else{
                        qu.push_back(v);//插入队尾 
                    }
                    inq[v]=1;
                    cnt[v]++;
                    if(cnt[v]>n){
                        return 0;   
                    }
                }
            }
        }
    }
    return 1;
}


int main(){
    cin >> n >> m;
     for(int i=1;i<=n;i++){
        add(0,i,0);
    }
    for(int i=0;i<m;i++){
        int d,a,b,c;
        cin >> d;
        if(d==1){//连权值 
            cin >>a>>b>>c;
            g[a].push_back(node(b,-c));
        }else if(d==2){
            cin >>a>>b>>c;
            g[b].push_back(node(a,c));
        }else{
            cin >>a>>b;
            g[a].push_back(node(b,0));
            g[b].push_back(node(a,0));
        }
    }
    
   
    if(spfa(0)){
        cout << "Yes";
    }else{
        cout << "No";
    }
    return 0;
}

P1993 小K的农场(差分约束)

原文:https://www.cnblogs.com/A-Konnyaku/p/10993063.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!