zoukankan      html  css  js  c++  java
  • CF1076D Edge Deletion

    洛谷传送门
    cf传送门
    这道题作为div.2的D题,被我一眼秒了我觉得十分荣幸,然后就开始写,然后就写了好久。
    AC之后看网上的题解,发现好多最短路树的,猛然发现我写的好复杂啊,结果还看到了直接一遍dijkstra+贪心的标算。。。
    只有我建的最短路图,再慢慢删边的吗。。。
    说白了也是贪心啦,
    策略十分显然:
    1、如果有多条最短路到一个点,显然可以删到只剩一条,这也是最先需要做的(最短路树的不用管这个)
    2、然后就只剩下一颗树了,topsort即可
    代码:

    #include<cstdio>
    #include<queue>
    #include<cstring>
    #include<vector>
    #include<algorithm>
    using namespace std;
    #define rg register
    vector<int>s[300001];
    long long dis[300001];int n,m,k,cnt,sum,x[300001],used[300001],y[300001],in[300001],out[300001],z[300001],pre[600001],nxt[600001],h[300001],v[600001],op[600001];
    struct oo{int id;long long dis;};bool vis[300001];
    bool operator<(oo a,oo b){return a.dis>b.dis;}
    priority_queue<oo>q;
    void add(int x,int y,int z,int now)
    {
    	pre[++cnt]=y,nxt[cnt]=h[x];h[x]=cnt,v[cnt]=z;op[cnt]=now;
    	pre[++cnt]=x,nxt[cnt]=h[y];h[y]=cnt,v[cnt]=z;op[cnt]=now;
    }
    void ins(int x,int y,int z,int now){pre[++cnt]=y,nxt[cnt]=h[x];h[x]=cnt,v[cnt]=z;op[cnt]=now;}
    void dijkstra()
    {
    	memset(dis,63,sizeof dis);
    	q.push((oo){1,dis[1]=0});
    	while(!q.empty())
    	{
    		int x=q.top().id;q.pop();
    		if(vis[x])continue;vis[x]=1;
    		for(int i=h[x];i;i=nxt[i])
    			if(dis[pre[i]]>dis[x]+v[i])dis[pre[i]]=dis[x]+v[i],q.push((oo){pre[i],dis[pre[i]]});
    	}
    }
    bool cmp(vector<int>a,vector<int>b){return a.size()>b.size();}
    void topsort()
    {
    	queue<int>q;
    	for(int i=1;i<=n;i++)if(!in[i])q.push(i);
    	while(!q.empty())
    	{
    		int x=q.front();q.pop();
    		for(int i=h[x];i;i=nxt[i])
    		{
    			used[op[i]]=0,sum--;if(sum<=k)return ;
    			if(!(--in[pre[i]]))q.push(pre[i]);
    		}
    	}
    }
    int main()
    {
    	scanf("%d%d%d",&n,&m,&k);
    	for(rg int i=1;i<=m;i++)scanf("%d%d%d",&x[i],&y[i],&z[i]),add(x[i],y[i],z[i],i);
    	dijkstra();
    	for(int i=1;i<=n;i++)
    		for(int j=h[i];j;j=nxt[j])
    			if(dis[pre[j]]==dis[i]+v[j])used[op[j]]=i;
    	memset(h,0,sizeof h),cnt=0;
    	for(int i=1;i<=m;i++)
    		if(used[i])
    		{
    			sum++;
    			if(used[i]==x[i])s[y[i]].push_back(i),in[y[i]]++,out[x[i]]++;
    			else s[x[i]].push_back(i),in[x[i]]++,out[y[i]]++;
    		}
    	if(sum<=k)
    	{
    		printf("%d
    ",sum);
    		for(int i=1;i<=m;i++)if(used[i])printf("%d ",i);
    		return 0;
    	}
    	int now=1;
    	while(now<=n)
    	{
    		if(s[now].size()>1)
    		{
    			int w=s[now].size(),t;
    			if(w-1>=sum-k)t=sum-k;
    			else t=w-1;sum-=t;
    			for(int i=0;i<t;i++)used[s[now][i]]=0;
    			if(sum<=k)break;
    		}
    		if(sum<=k)break;
    		now++;
    	}
    	if(sum<=k)
    	{
    		printf("%d
    ",sum);
    		for(int i=1;i<=m;i++)if(used[i])printf("%d ",i);
    		return 0;
    	}
    	memset(in,0,sizeof in),sum=cnt=0,memset(h,0,sizeof h);
    	for(int i=1;i<=m;i++)
    		if(used[i])
    		{
    			sum++;
    			if(used[i]==x[i])ins(y[i],x[i],z[i],i),in[x[i]]++;
    			else ins(x[i],y[i],z[i],i),in[y[i]]++;
    		}
    	topsort();
    	printf("%d
    ",sum);
    	for(int i=1;i<=m;i++)if(used[i])printf("%d ",i);
    }
    
  • 相关阅读:
    树上差分
    Java学习笔记(二)事件监听器
    Java学习笔记(三)Java2D组件
    1066. Root of AVL Tree (25)
    有一种蓝,是神往,是心醉,是心伤
    软考论文的六大应对策略V1.0
    iOS 图形编程总结
    【悼鲁迅】诗一首
    【秋游】诗一首
    【游普罗旺斯薰衣草庄园】诗一首
  • 原文地址:https://www.cnblogs.com/lcxer/p/10028222.html
Copyright © 2011-2022 走看看