HYSBZ 1079(着色方案)

Dp 神奇的状态转移……


Program fd;
const
   mo=1000000007;
var
   n,i,j:longint;
   c,tot:array[1..15] of longint;
   f:array[0..15,0..15,0..15,0..15,0..15,0..15] of int64;
function dfs(a1,a2,a3,a4,a5,last:longint):int64;
var
   i,j:longint;
   ans,q:int64;
   a:array[1..5] of longint;
begin
   if f[a1,a2,a3,a4,a5,last]>0 then exit(f[a1,a2,a3,a4,a5,last]);
   a[1]:=a1;a[2]:=a2;a[3]:=a3;a[4]:=a4;a[5]:=a5;
   ans:=0;
   for i:=1 to 5 do
      if (a[i]>0) then
      begin
         if (last=i+1) and (a[i]=1) then continue;

  //     if (last=i+1) then dec(a[i]);
         dec(a[i]);
         if i>1 then inc(a[i-1]);


         q:=dfs(a[1],a[2],a[3],a[4],a[5],i);
         inc(a[i]);
         if i>1 then dec(a[i-1]);

         if last=i+1 then ans:=(ans+(a[i]-1)*q) mod mo
         else ans:=(ans+a[i]*q) mod mo;


  //     if (last=i+1) then inc(a[i]);
      end;
   f[a1,a2,a3,a4,a5,last]:=ans;
   exit(ans);
end;
begin
   read(n);
   fillchar(tot,sizeof(tot),0);
   for i:=1 to n do
   begin
      read(c[i]);
      inc(tot[c[i]]);
   end;
   fillchar(f,sizeof(f),0);
   for i:=1 to n do f[0,0,0,0,0,i]:=1;
   writeln(dfs(tot[1],tot[2],tot[3],tot[4],tot[5],0));



end.


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
邻接矩阵是一种用于表示无向图的矩阵结构,其中矩阵的行和列均代表图中的顶点,如果两个顶点之间有边相连,则对应的矩阵元素为1,否则为0。在这个问题中,我们需要判断给定的无向图是否能够用3种颜色进行着色,具体方法如下: 1. 读入邻接矩阵 我们首先需要读入给定的无向图的邻接矩阵,假设矩阵的大小为V * V,其中V为顶点数。可以使用二维数组来表示邻接矩阵,例如: ``` int G[MAXV][MAXV]; // 图的邻接矩阵表示 int V; // 图的顶点数 // 读入图的顶点数和邻接矩阵 scanf("%d", &V); for (int i = 0; i < V; i++) { for (int j = 0; j < V; j++) { scanf("%d", &G[i][j]); } } ``` 2. 检查相邻顶点的颜色是否相同 在进行图的着色时,我们需要保证相邻的顶点颜色不同。因此,对于每个顶点v,需要检查与之相邻的所有顶点的颜色是否与v的颜色相同。如果相同,则说明这个顶点不能着当前的颜色。 ``` bool is_valid(int v, int c) { for (int i = 0; i < V; i++) { if (G[v][i] && color[i] == c) { return false; } } return true; } ``` 这个函数接受两个参数,分别是当前顶点的编号v和当前尝试的颜色c。它遍历与v相邻的所有顶点i,如果i与v相邻且颜色相同,则返回false,否则返回true。 3. 尝试给每个顶点着色着色时,我们可以使用回溯法来尝试给每个顶点着色。具体来说,我们从第一个顶点开始,依次尝试用三种颜色中的一种进行着色,如果当前的着色方案合法,则继续对下一个顶点进行着色。如果所有顶点都已经着色,则说明找到了一种合法的着色方案。 ``` bool backtrack(int v) { // 如果所有顶点都已经着色,返回true if (v == V) { return true; } // 尝试给当前顶点着色 for (int c = 1; c <= MAXC; c++) { if (is_valid(v, c)) { color[v] = c; // 继续给下一个顶点着色 if (backtrack(v + 1)) { return true; } // 如果不能继续给下一个顶点着色,回溯 color[v] = 0; } } // 如果无法给当前顶点着色,返回false return false; } ``` 这个函数接受一个参数v,表示当前需要进行着色的顶点编号。它首先检查是否所有顶点都已经着色,如果是,则返回true。否则,它尝试给当前顶点着三种颜色中的一种,检查是否合法,如果合法,则继续对下一个顶点进行着色;否则,回溯到上一个顶点,重新尝试其他颜色。 4. 输出着色方案 如果找到了一种合法的着色方案,则输出这个方案。否则,说明这个图不能用三种颜色进行着色满足相邻顶点颜色互异的要求。 ``` // 尝试给每个顶点着色,如果成功则输出着色方案 if (backtrack(0)) { printf("Yes\n"); for (int i = 0; i < V; i++) { printf("%d ", color[i]); } printf("\n"); } else { printf("No\n"); } ``` 这段代码调用backtrack函数,并根据返回值判断是否找到了合法的着色方案。如果找到了,输出"YES",并输出每个顶点的颜色编号;否则,输出"NO"。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值