POJ 3253 Fence Repair STL堆操作
我想说,STL里堆是我目前见到最蛋疼的操作。
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #include <cstdlib> #include <cmath> #include <utility> #include <vector> #include <queue> #include <map> #include <set> #define max(x,y) ((x)>(y)?(x):(y)) #define min(x,y) ((x)>(y)?(y):(x)) using namespace std; int n,a[20005]; int main() { scanf("%d",&n); for(int i=0; i<n; i++) scanf("%d",&a[i]); vector< int > v(a,a+n); make_heap(v.begin(),v.end(),greater< int >()); long long ans=0; for(int i=1; i<=n-1; i++) { long long tmp=v.front(); pop_heap(v.begin(),v.end(),greater< int >()); v.pop_back(); tmp+=v.front(); pop_heap(v.begin(), v.end(),greater< int >()); v.pop_back(); v.push_back(tmp); push_heap(v.begin(),v.end(),greater< int >()); ans+=tmp; printf("%d\n", tmp); } printf("%lld\n",ans); return 0; }
POJ 3254 Corn Fields 状态压缩递推
uc[]表示相邻两位不同时为1的情况,377种,因此n*377^2即可
注意,位运算优先级是低于==的!
#include <cstdio> #include <cstring> #define MOD 100000000 int n,m,a[20],k; int uc[500],all; int f[20][5000]; int main() { scanf("%d%d",&n,&m); for(int i=0; i<n; i++) for(int j=0; j<m; j++) { scanf("%d",&k); a[i]=(a[i]<<1)+k; } for(int i=0; i<(1<<m); i++) { int tmi=i,flag=0; while(tmi) { if((tmi&3)==3) flag=1; tmi>>=1; } if(!flag) { uc[all++]=i; if((i|a[0])==a[0]) f[0][i]=1; } } for(int i=1; i<n; i++) for(int j=0; j<all; j++) if((a[i]|uc[j])==a[i]) { for(int k=0; k<all; k++) if(!(uc[j]&uc[k])) { f[i][uc[j]]=(f[i][uc[j]]+f[i-1][uc[k]])%MOD; } } int ans=0; for(int j=0; j<all; j++) ans=(ans+f[n-1][uc[j]])%MOD; printf("%d\n", ans); return 0; }
POJ 3255 A*算法求第k短路
第一次接触A*,了解了下。
—————————————————维基百科————————————————————————————
如果以 g(n)表示从起点到任意顶点n的实际距离,h(n)表示任意顶点n到目标顶点的估算距离,那么 A*算法的公式为:f(n)=g(n)+h(n)。 这个公式遵循以下特性:
- 如果h(n)为0,只需求出g(n),即求出起点到任意顶点n的最短路径,则转化为单源最短路径问题,即Dijkstra算法
- 如果h(n)<=“n到目标的实际距离”,则一定可以求出最优解。而且h(n)越小,需要计算的节点越多,算法效率越低。
—————————————————————————————————————————————————
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #include <cstdlib> #include <cmath> #include <utility> #include <vector> #include <queue> #include <map> #include <set> #define max(x,y) ((x)>(y)?(x):(y)) #define min(x,y) ((x)>(y)?(y):(x)) #define INF 0x3f3f3f3f using namespace std; struct Edge { int y,w,ne; }e[200005]; int x,y,w,n,m; int be[5005],all; int h[5005]; bool vis[5005]; struct Point { int x,g; bool operator < (const Point T) const { return g+h[x]>T.g+h[T.x]; } }; void add(int x, int y, int w) { e[all].y=y; e[all].w=w; e[all].ne=be[x]; be[x]=all++; } void SPFA(int s) { queue< int > q; for(int i=0; i<=n; i++) { h[i]=INF; vis[i]=0; } h[s]=0; vis[s]=1; while(!q.empty()) q.pop(); q.push(s); while(!q.empty()) { int u=q.front(); q.pop(); vis[u]=0; for(int i=be[u]; i!=-1; i=e[i].ne) { int v=e[i].y; if(h[v]>h[u]+e[i].w) { h[v]=h[u]+e[i].w; if(!vis[v]) { vis[v]=1; q.push(v); } } } } } int Astar(int s, int t, int k) { SPFA(t); priority_queue< Point > q; while(!q.empty()) q.pop(); memset(vis,0,sizeof(vis)); Point cur,nxt; cur.x=s; cur.g=0; q.push(cur); while(!q.empty()) { cur=q.top(); q.pop(); if(cur.x==t && !--k) return cur.g; for(int i=be[cur.x]; i!=-1; i=e[i].ne) { nxt.x=e[i].y; nxt.g=cur.g+e[i].w; q.push(nxt); } } return -1; } void init() { all=0; memset(be,-1,sizeof(be)); } int main() { scanf("%d%d",&n,&m); init(); for(int i=0; i<m; i++) { scanf("%d%d%d",&x,&y,&w); add(x,y,w); add(y,x,w); } printf("%d\n",Astar(1,n,2)); return 0; }
时间: 2024-10-07 09:57:02