JZOJ 1843——小球

---

这题，只有一种情况才可以加，就是c[i]<>c[j]
那么定义一个sum1和sum2为1和0分别出现多少次
每次循环开始s1+sum1,s2+sum2
如果为0，则ans+s1,sum1+1
如果为1，则ans+s2,sum2+1

---

代码如下:

var n,i,x:longint;
    ans,s1,s2,sum1,sum2:int64;

begin
  readln(n);
  for i:=1 to n do
    begin
      read(x);
      inc(s1,sum1); inc(s2,sum2);
      case x of
        0:begin
            ans:=ans+s2;
            inc(sum1);
          end;
        1:begin
            ans:=ans+s1;
            inc(sum2);
          end;
      end;
    end;
  writeln(ans);
end.